Update app.py

app.py

@@ -1,21 +1,23 @@
 import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
-import random
 
 model_id = "nitrosocke/mo-di-diffusion"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cpu")
+local_dir = "./model/"
+
+# Download the model from the internet and save it to local directory
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, cache_dir=local_dir)
+
+# Load the model from the local directory on subsequent runs
+pipe = StableDiffusionPipeline.from_pretrained(local_dir, torch_dtype=torch.float16)
 
 def generate_image(prompt):
-    random.seed(42)  # set seed for reproducibility
-    prompt = f"modern disney style, {prompt}, seed={random.randint(0, 100000)}"
     image = pipe(prompt).images[0]
     image.save("./magical_princess.png")
     return image
 
 inputs = gr.inputs.Textbox(label="Enter Prompt")
-outputs = gr.outputs.Image(label="Generated Image")
+outputs = gr.outputs.Image(label="Generated Image", type="pil")
 
 title = "Midjourney-Disney"
 description = None