changed model_name

main.py

@@ -26,7 +26,7 @@ def generate_image(prompt, model):
     modelVersion = modelArray[1]
 
     pipeline = StableDiffusionPipeline.from_pretrained(
-        str(model), torch_dtype=torch.float16
+        str(modelName), torch_dtype=torch.float16
     )
     pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
     pipeline = pipeline.to("cuda")