Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 
 model_name_or_path = "lyttt/VLV_captioner"
 model = AutoModel.from_pretrained(model_name_or_path, revision="master", trust_remote_code=True,low_cpu_mem_usage=False)
+model = model.to("cuda")
 
 def drop_incomplete_tail(text):
     sentences = text.split('.')