Update app.py

app.py

@@ -43,10 +43,12 @@ with gr.Blocks(css="custom.css") as demo:
         token = os.getenv("API_TOKEN")  # This will read the API token set in your managed environment
         payload = {"inputs": f"{prompt} {trigger_word}"}
         
+        print("Calling query function...")
         image_bytes = query(payload, api_url, token)
+        print("Query function executed successfully.")
         return Image.open(image_bytes)
 
-
+    print("Starting Gradio UI...")
     prompt.submit(
         fn=run_lora,
         inputs=[prompt, weight, selected_state],