Update app.py

app.py

@@ -11,9 +11,9 @@ def generate_caption_clipgpt(image):
     caption = clipGPT.generate_caption_clipgpt(image)
     return caption
 
-def generate_caption_vitgpt(image):
-    caption = vitGPT.generate_caption(image)
-    return caption
+# def generate_caption_vitgpt(image):
+#     caption = vitGPT.generate_caption(image)
+#     return caption
 
 # Sample image paths
 sample_images = [
@@ -40,8 +40,8 @@ with gr.Blocks() as demo:
     def predict(img, model_name):
         if model_name == "CLIP-GPT2":
             return generate_caption_clipgpt(img)
-        elif model_name == "ViT-GPT2":
-            return generate_caption_vitgpt(img)
+        # elif model_name == "ViT-GPT2":
+        #     return generate_caption_vitgpt(img)
         else:
             return "Caption generation for this model is not yet implemented."