Update app.py

app.py

@@ -1,25 +1,46 @@
-from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
-import torch
-
-model_id = "stabilityai/stable-diffusion-2"
-
-# Use the Euler scheduler here instead
-scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
-
-prompt = "realistic photo of 3d cartoon style of a cat in a space suit with milky way background"
-
-def txt2img(prompt):
-
-  image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0]
-    
-  image.save("sd_image.png")
-
-  return image
-
-txt2img(prompt)
-
-import gradio as gr 
-
-gr.Interface(txt2img, gr.Text(), gr.Image(), title = 'Stable Diffusion 2.0 Colab with Gradio UI').launch(share = True, debug = True)
+import gradio as gr
+import gradio.components as comp
+import os
+
+api_key = os.environ.get("HUGGINGFACE_API_KEY")
+
+#model_list = [
+#    "stabilityai/stable-diffusion-xl-base-0.9",
+#    "stabilityai/stable-diffusion-2-1",
+#    "stabilityai/stable-diffusion-xl-refiner-0.9",
+#    "stabilityai/stable-diffusion-2-1-base",
+#    "stabilityai/stable-diffusion-2",
+#    "stabilityai/stable-diffusion-2-inpainting",
+#    "stabilityai/stable-diffusion-x4-upscaler",
+#    "stabilityai/stable-diffusion-2-depth",
+#    "stabilityai/stable-diffusion-2-base",
+#    "stabilityai/stable-diffusion-2-1-unclip",
+#    "helenai/stabilityai-stable-diffusion-2-1-base-ov",
+#    "helenai/stabilityai-stable-diffusion-2-1-ov",
+#    "stabilityai/stable-diffusion-2-1-unclip-small"
+#]
+
+#default_model = "stabilityai/stable-diffusion-2"
+#model_name = gr.inputs.Dropdown(choices=model_list, label="Select Model", default=default_model)
+
+#def generate_image(text, default_model):
+#    model = gr.load(default_model, source="huggingface", api_key=api_key)
+#    return model.predict(text)
+
+#input_text = gr.inputs.Textbox(label="Input Text")
+#output_image = comp.Image(label="Generated Image")
+
+#iface = gr.Interface(
+#    fn=generate_image,
+#    inputs=[input_text, default_model],
+#    outputs=output_image,
+#    title="Text to Image Generation",
+#    description="Generate an image from input text using a Hugging Face model."
+#)
+
+#iface.launch()
+
+title = ".mujo T ai art po O"
+gr.Interface.load("models/stabilityai/stable-diffusion-2-1", title=title).launch()
+
+#gr.load("models/stabilityai/stable-diffusion-2-1-base").launch(auth=("admin", "pass"))