Add stuff

.gitignore

@@ -0,0 +1 @@
+flagged/

.ipynb_checkpoints/.gitignore-checkpoint

@@ -0,0 +1 @@
+flagged/

.ipynb_checkpoints/README-checkpoint.md

@@ -0,0 +1,12 @@
+---
+title: Stablepod
+emoji: 🌖
+colorFrom: pink
+colorTo: red
+sdk: gradio
+sdk_version: 3.4
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

.ipynb_checkpoints/app-checkpoint.py

@@ -0,0 +1,18 @@
+import gradio as gr
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
+
+
+def get_stable_diffusion_random_image(prompt):
+    model_id = "CompVis/stable-diffusion-v1-4"
+    device = "cuda"
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
+    pipe = pipe.to(device)
+    with autocast("cuda"):
+        image = pipe(prompt, guidance_scale=7.5).images[0]  
+        return image
+
+
+iface = gr.Interface(fn=get_stable_diffusion_random_image, inputs="text", outputs="image")
+iface.launch(share=True)

app.py

@@ -1,7 +1,18 @@
 import gradio as gr
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def get_stable_diffusion_random_image(prompt):
+    model_id = "CompVis/stable-diffusion-v1-4"
+    device = "cuda"
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
+    pipe = pipe.to(device)
+    with autocast("cuda"):
+        image = pipe(prompt, guidance_scale=7.5).images[0]  
+        return image
+
+
+iface = gr.Interface(fn=get_stable_diffusion_random_image, inputs="text", outputs="image")
+iface.launch(share=True)