Update app.py

app.py

@@ -1,3 +1,26 @@
+import os 
+import huggingface_hub as hf_hub
 import gradio as gr
 
-gr.load("models/stabilityai/stable-diffusion-xl-base-1.0").launch()
+client = hf_hub.InferenceClient(token = os.environ['HF_TOKEN'])
+
+def image_interface(prompt, negative_prompt, guidance_scale):
+    response = client.text_to_image(
+        prompt = prompt,
+        negative_prompt = negative_prompt,
+        model = 'stabilityai/stable-diffusion-xl-base-1.0',
+        guidance_scale = guidance_scale,
+        num_inference_steps = 50, 
+    )
+
+    return response
+
+app = gr.Interface(
+    fn = image_interface, 
+    inputs = [gr.Textbox(label = 'Prompt'), gr.Textbox(label = 'Negative Prompt'), gr.Slider(minimum = 1, maximum = 30, value = 7, step = 0.5, label = 'Guidance Scale', show_label = True)], 
+    outputs = 'image', 
+    title = 'Stable Diffusion',
+    description = 'Vinay Kumar Thakur'
+)
+
+app.launch()