Update app.py

app.py

@@ -52,7 +52,7 @@ print("F1 Score:", f1_score(test_labels, distilbert_predictions, average='weight
 # Function to generate an image from a prompt using Stable Diffusion
 def generate_image(prompt):
     try:
-        pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", variant="fp32", torch_dtype=torch.float32)#.to("cuda")
+        pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", variant="fp16", torch_dtype=torch.float32)#.to("cuda")
         output = pipe(prompt=prompt)
         image = output.images[0]
         image.save("output_image.png")