Update app.py

app.py

@@ -54,17 +54,18 @@ async def generate_image(
     text = str(translator.translate(prompt, 'English')) + "," + lora_word
 
     client = AsyncInferenceClient()
-
-    image = await client.text_to_image(
-        prompt=text,
-        negative_prompt=negative_prompt,
-        height=height,
-        width=width,
-        guidance_scale=scales,
-        num_inference_steps=steps,
-        model=model,
-    )
-    print(image)
+    try:
+        image = await client.text_to_image(
+            prompt=text,
+            negative_prompt=negative_prompt,
+            height=height,
+            width=width,
+            guidance_scale=scales,
+            num_inference_steps=steps,
+            model=model,
+        )
+    except Exception as e:
+        raise gr.Error(f"Error in {e}")
     
     return image, seed