Spaces:

merve
/

compare_VLMs

Runtime error

merve HF staff commited on Apr 23

Commit

b809dba

•

1 Parent(s): 9bc4911

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -121,7 +121,7 @@ def generate_caption_llava(image_path, caption_bool):
     inputs = processor(text_prompt, Image.open(image_path), return_tensors="pt").to(device)
     output = model.generate(**inputs, max_new_tokens=100)
-    return processor.decode(output[0], skip_special_tokens=True)).split("[/INST]")[1]
 @spaces.GPU
 def generate_answer_llava(image_path, question):

     inputs = processor(text_prompt, Image.open(image_path), return_tensors="pt").to(device)
     output = model.generate(**inputs, max_new_tokens=100)
+    return processor.decode(output[0], skip_special_tokens=True).split("[/INST]")[1]
 @spaces.GPU
 def generate_answer_llava(image_path, question):