Update app.py

app.py

@@ -105,10 +105,10 @@ def predict_event(image):
 
     # caption_vitgpt = generate_caption(model, image)
     #caption_vitgpt = generate_caption(feature_extractor, cap_model, image, tokenizer)
-    preds = [pred.strip() for pred in generated_caption]
-    return preds
+    # preds = [pred.strip() for pred in generated_caption]
+    # return preds
 
-    #return '\033[96m' +generated_caption[:85]+ '\033[0m'
+    return '\033[96m' +generated_caption[:85]+ '\033[0m'