Update app.py

app.py

@@ -21,10 +21,10 @@ def get_caption(img):
     return output[0]["generated_text"]
 
 # 4. Image Generation
-img_model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+# img_model = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+img_model = DiffusionPipeline.from_pretrained("Linaqruf/anything-v3.0")
 def get_img(prompt):
     return img_model(prompt).images[0]
-    
 
 demo = gr.Blocks()
 with demo: