Update app.py

app.py

@@ -114,7 +114,11 @@ def model_generate_ans(img=None,img_audio=None,val_q=None):
             
         predicted_captions_decoded = tokenizer.batch_decode(predicted_caption,ignore_index = 50256)[0]
     
-    return predicted_captions_decoded
+    # Split the string at the first occurrence of <|endoftext|>
+    result = predicted_captions_decoded.split('<|endoftext|>')[0]
+    return result.strip()  # Strip any trailing spaces or newlines
+    
+    #return predicted_captions_decoded
     
 
 with gr.Blocks() as demo: