Update app.py

app.py

@@ -235,8 +235,8 @@ if uploaded_file is not None and analyze_button:
         if isinstance(caption_response, dict) and caption_response.get("error"):
             st.error(f"Error with image captioning model: {caption_response['error']}")
         else:
-            # Extract the generated caption
-            image_caption = caption_response[0]['generated_text']
+            # Since caption_response is a string, assign it directly
+            image_caption = caption_response
 
             # Use the complex image prompt text
             prompt_text = complex_image_prompt_text