updates

app.py

@@ -14,7 +14,7 @@ def generate(inp):
     torch.cuda.empty_cache()
     print(f"Is CUDA available: {torch.cuda.is_available()}")
 
-    pipeline = StableDiffusionPipeline.from_pretrained(str(model), torch_dtype=torch.float16)
+    pipeline = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
     
     #pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
     #another comment

main.py

@@ -20,7 +20,7 @@ def generate_image(prompt, inference_steps, model):
     torch.cuda.empty_cache()
     print(f"Is CUDA available: {torch.cuda.is_available()}")
 
-    pipeline = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype=torch.float16)
+    pipeline = StableDiffusionPipeline.from_pretrained(str(model), torch_dtype=torch.float16)
     
     #pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
     #another comment