Update app.py

app.py

@@ -87,9 +87,11 @@ def process_image(prompt: str, img: Image.Image) -> str:
 print("Diffusion path: ", diffusion_path)
 # Initialize model + pipeline
 disable_torch_init()
+
 tokenizer, multi_model, _ = load_pretrained_model(
-    diffusion_path, None, get_model_name_from_path(diffusion_path)
+    model_snapshot_path, None, get_model_name_from_path(model_snapshot_path)
 )
+
 pipe = DiffusionPipeline.from_pretrained(
     diffusion_path,
     custom_pipeline="pipeline_llava_gen",