Update app.py

app.py

@@ -1,11 +1,8 @@
-import os
+from diffusers import StableDiffusionPipeline
+import torch
 import gradio as gr
 
-API_KEY=os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
-article = """---
-This space was created using [SD Space Creator](https://huggingface.co/spaces/anzorq/sd-space-creator)."""
-
-models = [
+models=[
 "models/ItsJayQz/Marvel_WhatIf_Diffusion",
 "models/DGSpitzer/Cyberpunk-Anime-Diffusion",
 "models/DGSpitzer/Guan-Yu-Diffusion",
@@ -16,22 +13,17 @@ models = [
 "models/stabilityai/stable-diffusion-2-1"
 ]
 
-custom_model = "models/dreamlike-art/dreamlike-diffusion-1.0"
-
-def selectModel(message, models):
-    message = message.lower()
-    return models
-
-sandbox = gr.Interface.load(
-    fn= selectModel,
-    title="AlStable sandbox text to img",
-    name = models[3], 
-    inputs = ["text", gr.Dropdown(models)],
-    outputs = "image", 
-    description="Demo for <a href='https://huggingface.co/stabilityai/stable-diffusion-2-1'>AlStable</a> Stable Diffusion model.",
-    article=article,
-    api_key=API_KEY
-)
+def TextToImage(Prompt,model):
+  model_id = model
+  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+  pipe = pipe.to("cpu")
+  prompt = Prompt
+  image = pipe(prompt).images[0]
+  return image
 
-sandbox.queue(concurrency_count=20).launch()
+sandbox = gr.Interface(fn=TextToImage, 
+                        inputs=["text", gr.Dropdown(models)],
+                         outputs="image", 
+                        title='AlStable Text to Image')
 
+sandbox.launch()