commit

app.py

@@ -29,14 +29,17 @@ def generate_caption_and_analyze_emotions(image):
     # Get the predicted emotion label
     emotion_label = emotion_tokenizer.decode(emotion_outputs.logits.argmax())
 
-
     # Prepare the final output with sentiment information
-    #final_output = f"The sentiment in the provided image shows: {emotion_label}.\n\nGenerated Caption: {decoded_caption}"
-    return emotion_label
+    final_output = f"The sentiment in the provided image shows: {emotion_label}.\n\nGenerated Caption: {decoded_caption}"
+    return final_output
 
 # Define the Gradio interface
 inputs = gr.inputs.Image(label="Upload an image")
-outputs = gr.outputs.Textbox(label="Generated Sentiment")
+outputs = gr.outputs.Textbox(label="Generated Caption and Sentiment Analysis")
 
 # Create the Gradio app
-gr.Interface(fn=generate_caption_and_analyze_emotions, inputs=inputs, outputs=outputs).launch()
+app = gr.Interface(fn=generate_caption_and_analyze_emotions, inputs=inputs, outputs=outputs)
+
+# Launch the Gradio app
+if __name__ == "__main__":
+    app.launch()