Update app.py

app.py

@@ -53,16 +53,16 @@ caption = gr.Textbox(label="Generated Caption")
     
 def predict(img, model_name, max_tokens, temperature):
     if model_name == "CLIP-GPT2":
-      return generate_caption_clipgpt(img, max_tokens, temperature)
+        return generate_caption_clipgpt(img, max_tokens, temperature)
     elif model_name == "ViT-GPT2":
-      return generate_caption_vitgpt(img, max_tokens, temperature)
+        return generate_caption_vitgpt(img, max_tokens, temperature)
     elif model_name == "ViT-CoAttention":
-      return generate_caption_vitCoAtt(img)
+        return generate_caption_vitCoAtt(img)
     elif model_name == "Baseline Model CNN-RNN":
         print(img)
-      return generate_caption_cnnrnn(img)
+        return generate_caption_cnnrnn(img)
     else:
-      return "Caption generation for this model is not yet implemented."   
+        return "Caption generation for this model is not yet implemented."