Update app.py

app.py

@@ -41,7 +41,7 @@ def generate_answer_git(processor, model, image, question):
     generated_ids = model.generate(pixel_values=pixel_values, input_ids=input_ids, max_length=128)#50)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
-    return generated_answer.replace(question, '').replace(question.lower(), '').strip()
+    return generated_answer[0].replace(question, '').replace(question.lower(), '').strip()
 
 
 def generate_answer_blip(processor, model, image, question):
@@ -51,7 +51,7 @@ def generate_answer_blip(processor, model, image, question):
     generated_ids = model.generate(**inputs, max_length=128)#50)
     generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
    
-    return generated_answer
+    return generated_answer[0].strip()
 
 
 def generate_answer_vilt(processor, model, image, question):
@@ -63,7 +63,7 @@ def generate_answer_vilt(processor, model, image, question):
 
     predicted_class_idx = outputs.logits.argmax(-1).item()
     
-    return model.config.id2label[predicted_class_idx]
+    return model.config.id2label[predicted_class_idx]#[0].strip()
 
 
 def generate_answers(image, question):