Update app.py

app.py

@@ -24,12 +24,19 @@ def generate_compliment(image):
     image_bytes = buffered.getvalue()
     
     # Connect to the captioning model on Hugging Face Spaces
-    captioning_url = "https://huggingface.co/spaces/gokaygokay/SD3-Long-Captioner"
-    caption_response = requests.post(captioning_url, files={"image": image_bytes})
-    caption = caption_response.json()["data"][0]
+    captioning_url = "https://gokaygokay-sd3-long-captioner.hf.space/run/create_captions_rich"
+    caption_response = requests.post(captioning_url, files={"image": ("image.jpg", image_bytes, "image/jpeg")})
+    
+    if caption_response.status_code != 200:
+        return "Error", f"Failed to get caption. Status code: {caption_response.status_code}, Response: {caption_response.text}"
+    
+    try:
+        caption = caption_response.json()["data"][0]
+    except Exception as e:
+        return "Error", f"Failed to parse caption response. Error: {str(e)}, Response: {caption_response.text}"
     
     # Connect to the LLM model on Hugging Face Spaces
-    llm_url = "https://huggingface.co/spaces/hysts/zephyr-7b"
+    llm_url = "https://hysts-zephyr-7b.hf.space/run/chat"
     llm_payload = {
         "system_prompt": SYSTEM_PROMPT,
         "message": f"Caption: {caption}\nCompliment: ",
@@ -40,7 +47,14 @@ def generate_compliment(image):
         "repetition_penalty": 1,
     }
     llm_response = requests.post(llm_url, json=llm_payload)
-    compliment = llm_response.json()["data"][0]
+    
+    if llm_response.status_code != 200:
+        return "Error", f"Failed to get compliment. Status code: {llm_response.status_code}, Response: {llm_response.text}"
+    
+    try:
+        compliment = llm_response.json()["data"][0]
+    except Exception as e:
+        return "Error", f"Failed to parse LLM response. Error: {str(e)}, Response: {llm_response.text}"
 
     return caption, compliment
 
@@ -53,7 +67,8 @@ iface = gr.Interface(
         gr.Textbox(label="Compliment")
     ],
     title="Compliment Bot 💖",
-    description="Upload your headshot and get a personalized compliment!"
+    description="Upload your headshot and get a personalized compliment!",
+    share=True
 )
 
 iface.launch()