commit

app.py

@@ -22,16 +22,19 @@ def generate_caption_and_analyze_emotions(image):
     emotion_classifier = pipeline(model=emotion_model_name)
 
     results = emotion_classifier(decoded_caption)
-    if results[0]['label'] == 'neutral' or results[0]['score'] <= 0.40:
-        final_output = f"Sentiment of image is not clear, image shows {decoded_caption}."
+    sentiment_label = results[0]['label']
+    if sentiment_label == 'neutral':
+        sentiment_text = "Sentiment of the image is"
     else:
-        final_output = f"Sentiment of the image shows {results[0]['label']}."
+        sentiment_text = "Sentiment of the image shows"
+
+    final_output = f"This image shows {decoded_caption} and {sentiment_text} {sentiment_label}."
 
     return final_output
 
 # Define the Gradio interface
 inputs = gr.inputs.Image(label="Upload an image")
-outputs = gr.outputs.Textbox(label="Sentiment Analysis")
+outputs = gr.outputs.Textbox(label="Generated Caption and Sentiment Analysis")
 
 # Create the Gradio app
 app = gr.Interface(fn=generate_caption_and_analyze_emotions, inputs=inputs, outputs=outputs)