Update app.py

app.py

@@ -84,11 +84,11 @@ def process_image(prompt: str, img: Image.Image) -> str:
     )[0]
     return output_text
 
+print("Diffusion path: ", diffusion_path)
 # Initialize model + pipeline
 disable_torch_init()
-model_path = os.path.expanduser(sys.argv[1])
 tokenizer, multi_model, _ = load_pretrained_model(
-    model_path, None, get_model_name_from_path(model_path)
+    model_path, None, get_model_name_from_path(diffusion_path)
 )
 pipe = DiffusionPipeline.from_pretrained(
     diffusion_path,