Update app.py

app.py

@@ -39,7 +39,9 @@ def run_lora(prompt, selected_state, progress=gr.Progress(track_tqdm=True)):
     if response.status_code == 200:
         return Image.open(io.BytesIO(response.content))
     else:
-        return "API Error"
+        logging.error(f"API Error: {response.status_code}")
+        raise gr.Error("API Error: Unable to fetch the image.")  # Raise a Gradio error here
+
 
 with gr.Blocks(css="custom.css") as app:
     title = gr.HTML("<h1>LoRA the Explorer</h1>")