Update app.py

app.py

@@ -16,7 +16,7 @@ model.device
 model_id = 'prompthero/midjourney-v4-diffusion' #"stabilityai/stable-diffusion-2"
 
 scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16) #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
+pipe = StableDiffusionPipeline.from_pretrained(model_id) #pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16 #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 
 def transcribe(audio,prompt_num,user_keywords):
@@ -173,8 +173,8 @@ def keywords(text,prompt_num,user_keywords):
             count += 1
             print(i)
             print("works4")
-            with torch.autocast("cuda"):
-                image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
+            # with torch.autocast("cuda"):
+            image = pipe(i, height=768, width=768, guidance_scale = 10).images[0]
             print("works5")
             images.append(image)
             print("works6")