Update app.py

app.py

@@ -13,6 +13,7 @@ from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 import requests
 import logging
+from fastapi import FastAPI, HTTPException
 
 # Set up logging to suppress print statements in UI
 logging.basicConfig(level=logging.INFO, filename='qwen_image_editor.log', filemode='a')
@@ -68,9 +69,11 @@ scheduler_config = {
 scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
 
 # Load the model pipeline
-pipe = QwenImageEditPlusPipeline.from_pretrained("Qwen/Qwen-Image-Edit-2509", 
-                                                 scheduler=scheduler,
-                                                 torch_dtype=dtype).to(device)
+pipe = QwenImageEditPlusPipeline.from_pretrained(
+    "Qwen/Qwen-Image-Edit-2509", 
+    scheduler=scheduler,
+    torch_dtype=dtype
+).to(device)
 pipe.load_lora_weights(
     "lightx2v/Qwen-Image-Lightning", 
     weight_name="Qwen-Image-Lightning-4steps-V2.0.safetensors"
@@ -593,7 +596,18 @@ def create_demo():
 
     return demo
 
+# --- FastAPI Setup for Path Restriction ---
+app = FastAPI()
+
+# Mount Gradio app at /spaceishere
+demo = create_demo()
+app.mount("/spaceishere", demo.app)
+
+# Return 500 error for all other paths
+@app.get("/{path:path}")
+async def catch_all(path: str):
+    raise HTTPException(status_code=500, detail="Internal Server Error")
+
 if __name__ == "__main__":
     logger.info(f"Gradio version: {gr.__version__}")
-    demo = create_demo()
     demo.queue().launch(share=True)