Update app.py

app.py

@@ -15,6 +15,9 @@ from huggingface_hub import login
 # from prompt_augment import PromptAugment
 hf_token = os.getenv("hf")
 
+space_id = os.getenv("SPACE_ID")
+root_path = f"https://{space_id.replace('/', '-')}.hf.space" if space_id else ""
+
 if hf_token:
     print("Secret loaded successfully.")
 else:
@@ -124,6 +127,7 @@ ADAPTER_SPECS = {
 
 LOADED_ADAPTERS = set()
 LORA_OPTIONS = ["None"] + list(LORA_CONFIGS.keys()) + ["Lightning"]
+print("lora option: {LORA_OPTIONS[0]}")
 
 
 def load_lora(lora_name):
@@ -405,5 +409,5 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch(allowed_paths=["./"], ssr_mode=False, css=css)
+    demo.launch(allowed_paths=["./"], ssr_mode=False, css=css, root_path=root_path)