Update app.py

app.py

@@ -1,17 +1,15 @@
 import gradio as gr
-from diffusers import StableDiffusionPipeline
+from diffusers import DiffusionPipeline  
 import torch
 import os
 
-
-model_id = "Kotiko-ua/tryondiffusion-model"  
 token = os.getenv("HF_TOKEN")        
 
-pipe = StableDiffusionPipeline.from_pretrained(
-    model_id,
-    use_auth_token=token,            
-    torch_dtype=torch.float16
+pipe = DiffusionPipeline.from_pretrained(
+    "Kotiko-ua/tryondiffusion-model",
+    use_auth_token=token  # if the repo is gated
 )
+
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
 def try_on(person_img, cloth_img):