Update app.py

app.py

@@ -30,10 +30,8 @@ def generate_compliment(image):
         captioning_space = spaces.connect("gokaygokay/sd3-long-captioner")
         
         # Predict caption for the provided image
-        caption = captioning_space.predict("/create_captions_rich", { "image": image_bytes })
-        
-        # Extract the caption from the response
-        caption_text = caption.data[0]
+        caption_response = captioning_space.predict("/create_captions_rich", { "image": image_bytes })
+        caption_text = caption_response.data[0]
         
     except Exception as e:
         return "Error", f"Failed to get caption. Exception: {e}"
@@ -53,8 +51,8 @@ def generate_compliment(image):
             "repetition_penalty": 1,
         }
         
-        compliment = llm_space.run(llm_payload)
-        compliment_text = compliment.data[0]
+        compliment_response = llm_space.run(llm_payload)
+        compliment_text = compliment_response.data[0]
         
     except Exception as e:
         return "Error", f"Failed to generate compliment. Exception: {e}"
@@ -64,7 +62,7 @@ def generate_compliment(image):
 # Gradio interface setup
 iface = gr.Interface(
     fn=generate_compliment,
-    inputs=gr.inputs.Image(type="pil", label="Upload Image"),
+    inputs=gr.inputs.Image(label="Upload Image"),
     outputs=[
         gr.outputs.Textbox(label="Caption"),
         gr.outputs.Textbox(label="Compliment")