Update app.py

torch.get_default_dtype()

following discussion
https://discuss.huggingface.co/t/using-the-prompt-to-switch-model/30039/16

app.py

@@ -1,6 +1,7 @@
 from diffusers import StableDiffusionPipeline
 import torch
 import gradio as gr
+import accelerate
 
 models=[
 "ItsJayQz/Marvel_WhatIf_Diffusion",
@@ -15,7 +16,7 @@ models=[
 
 def TextToImage(Prompt,model):
   model_id = model
-  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.get_default_dtype())
   pipe = pipe.to("cpu")
   prompt = Prompt
   image = pipe(prompt).images[0]