Update app.py

app.py

@@ -112,11 +112,11 @@ def polish_prompt_hf(prompt, img_list):
         # Initialize the client
         prompt = f"{SYSTEM_PROMPT}\n\nUser Input: {prompt}\n\nRewritten Prompt:"
         client = InferenceClient(
-            provider="cerebras",
+            provider="nebius",
             api_key=api_key,
         )
 
-        image_url = None
+                image_url = None
         if img is not None:
             # If img is a PIL Image
             if hasattr(img, 'save'):  # Check if it's a PIL Image
@@ -177,8 +177,11 @@ def polish_prompt_hf(prompt, img_list):
             
         polished_prompt = polished_prompt.strip().replace("\n", " ")
         return polished_prompt
-    
-
+        
+    except Exception as e:
+        print(f"Error during API call to Hugging Face: {e}")
+        # Fallback to original prompt if enhancement fails
+        return original_prompt
 
 def encode_image(pil_image):
     import io