commit

app.py

@@ -29,13 +29,14 @@ 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 final_output
+    #final_output = f"The sentiment in the provided image shows: {emotion_label}.\n\nGenerated Caption: {decoded_caption}"
+    return emotion_label
 
 # Define the Gradio interface
 inputs = gr.inputs.Image(label="Upload an image")
-outputs = gr.outputs.Textbox(label="Generated Caption and Sentiment Analysis")
+outputs = gr.outputs.Textbox(label="Generated Sentiment")
 
 # Create the Gradio app
 gr.Interface(fn=generate_caption_and_analyze_emotions, inputs=inputs, outputs=outputs).launch()