Update app.py

app.py

@@ -25,10 +25,11 @@ def generate_compliment(image):
     
     # Connect to the captioning model on Hugging Face Spaces
     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_response = requests.get(captioning_url, files={"image": ("image.jpg", image_bytes, "image/jpeg")})
+        caption_response.raise_for_status()  # Raise an exception for HTTP errors
+    except requests.exceptions.RequestException as e:
+        return "Error", f"Failed to get caption. Exception: {e}"
     
     try:
         caption = caption_response.json()["data"][0]
@@ -46,10 +47,11 @@ def generate_compliment(image):
         "top_k": 50,
         "repetition_penalty": 1,
     }
-    llm_response = requests.post(llm_url, json=llm_payload)
-    
-    if llm_response.status_code != 200:
-        return "Error", f"Failed to get compliment. Status code: {llm_response.status_code}, Response: {llm_response.text}"
+    try:
+        llm_response = requests.post(llm_url, json=llm_payload)
+        llm_response.raise_for_status()  # Raise an exception for HTTP errors
+    except requests.exceptions.RequestException as e:
+        return "Error", f"Failed to get compliment. Exception: {e}"
     
     try:
         compliment = llm_response.json()["data"][0]
@@ -70,4 +72,4 @@ iface = gr.Interface(
     description="Upload your headshot and get a personalized compliment!"
 )
 
-iface.launch(share=True)
+iface.launch()