Update app.py

app.py

@@ -1,38 +1,13 @@
 
-# import gradio as gr
-# gr.Interface.load("models/faalbane/kopper-kreations-custom-sd-v-2-1-style-v2").launch() 
-
-#@title Install and import requirements
-!pip install -qqq diffusers==0.11.1 transformers gradio ftfy accelerate
-
-import diffusers
-import gradio
+import gradio as gr
 from PIL import Image
-def image_grid(imgs, rows, cols):
-    assert len(imgs) == rows*cols
-
-    w, h = imgs[0].size
-    grid = Image.new('RGB', size=(cols*w, rows*h))
-    grid_w, grid_h = grid.size
-    
-    for i, img in enumerate(imgs):
-        grid.paste(img, box=(i%cols*w, i//cols*h))
-    return grid
-
-#@title Load the model from the [Concepts Library](https://huggingface.co/sd-dreambooth-library). If you are new to Stable Diffusion, make sure you [read the LICENSE](https://github.com/CompVis/stable-diffusion/blob/main/LICENSE)
-#@markdown  You may also use a locally trained model by replacing the `model_id` to a path with the model locally or on Google Drive
-from torch import autocast
-from diffusers import StableDiffusionPipeline
+import diffusers
 import torch
+from diffusers import StableDiffusionPipeline
 
-model_id = "models/faalbane/kopper-kreations-custom-sd-v-2-1-model-object-copper-heart" #@param {type:"string"}
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
-
-#@title Run the Stable Diffusion pipeline with interactive UI Demo on Gradio
-#@markdown Run this cell to get a Gradio UI like this to run your models
-
-#@markdown ![](https://i.imgur.com/bxHfawQ.png)
-import gradio as gr
+# Load the model
+model_id = "faalbane/kopper-kreations-custom-sd-v-2-1-model-object-copper-heart"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda") 
 
 def inference(prompt, num_samples):
     all_images = [] 
@@ -41,18 +16,11 @@ def inference(prompt, num_samples):
     all_images.extend(images)
     return all_images
 
-with gr.Blocks() as demo:
-    gr.HTML("<h2 style=\"font-size: 2em; font-weight: bold\" align=\"center\">Stable Diffusion Dreambooth - Run Concept</h2>")
-    with gr.Row():
-        with gr.Column():
-            prompt = gr.Textbox(label="prompt")
-            samples = gr.Slider(label="Samples",value=1)
-            run = gr.Button(value="Run")
-        with gr.Column():
-            gallery = gr.Gallery(show_label=False)
-
-    run.click(inference, inputs=[prompt,samples], outputs=gallery)
-    # gr.Examples([["a photo of sks toy riding a bicycle", 1,1]], [prompt,samples], gallery, inference, cache_examples=False)
-
+# Create Gradio interface
+iface = gr.Interface(
+    fn=inference,
+    inputs=["textbox", "slider"],
+    outputs="gallery",
+)
 
-demo.launch(debug=True) 
+iface.launch()