description

app.py

@@ -30,7 +30,7 @@ While this current implementation may not support real-time processing, the pote
 def process_image_and_generate_output(image, model_selection):
     if image is None:
         return "Please select an image", None
-    if model_selection == 'Basic Model (Results won't be good)':
+    if model_selection == "Basic Model (Results won't be good)":
         result = inference_script.evaluate(image)
         pred_caption = ' '.join(result).rsplit(' ', 1)[0]
         pred_caption = pred_caption.replace('<unk>', '')
@@ -57,9 +57,9 @@ def process_image_and_generate_output(image, model_selection):
 
 sample_images = [
     [os.path.join(os.path.dirname(__file__), "sample_images/1.jpg"), "ViT-GPT2"],
-    [os.path.join(os.path.dirname(__file__), "sample_images/1.jpg"), 'Basic Model (Results won't be good)'],
+    [os.path.join(os.path.dirname(__file__), "sample_images/1.jpg"), "Basic Model (Results won't be good)"],
     [os.path.join(os.path.dirname(__file__), "sample_images/3.jpg"), "ViT-GPT2"],
-    [os.path.join(os.path.dirname(__file__), "sample_images/3.jpg"), 'Basic Model (Results won't be good)']
+    [os.path.join(os.path.dirname(__file__), "sample_images/3.jpg"), "Basic Model (Results won't be good)"]
 ]
 
 # Create a dropdown to select sample image