Update app.py

app.py

@@ -60,6 +60,9 @@ def generate(
 ):
     if torch.cuda.is_available():
 
+        if not use_img2img:
+            pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
+            
         if use_img2img:
             pipe = AutoPipelineForImage2Image.from_pretrained(model, torch_dtype=torch.float16)