Update app.py

app.py

@@ -33,17 +33,22 @@ except Exception as e:
     st.error(f"An error occurred: {e}")
 
 def generate_image_from_model(prompt):
-    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-    image_bytes = response.content
-    image = Image.open(io.BytesIO(image_bytes))
-    return image
-
+    try:
+        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+        image_bytes = response.content
+        image = Image.open(io.BytesIO(image_bytes))
+        return image
+    except Exception as e:
+        st.error(f"Error occurs:{e}")
+        
 def generate_image(prompt):
-    response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
-    image_bytes = response.content
-    image = Image.open(io.BytesIO(image_bytes))
-    return image
-
+    try:
+        response = requests.post(API_URL, headers=headers, json={"inputs": prompt})
+        image_bytes = response.content
+        image = Image.open(io.BytesIO(image_bytes))
+        return image
+    except Exception as e:
+        st.error(f"Error occurs:{e}")
 # Streamlit app
 def main():
     try: