Upload app.py

app.py

@@ -21,11 +21,11 @@ def generate_image(prompt, model_name):
 
 models = {
     "Stable Diffusion v1.5": "Yntec/stable-diffusion-v1-5",
-    "DreamBooth v2": "stabilityai/dreambooth-v2",
-    "DALL-E Mini": "dalle-mini/dalle-mini",
-    "DeepAI Generator": "deepai/image-gen",
-    "Realistic Vision": "realistic-vision/image-gen",
-    "Artistic AI": "artistic-ai/v2"
+    "DreamBooth v2": "Yntec/stable-diffusion-v1-5",
+    "DALL-E Mini": "Yntec/stable-diffusion-v1-5",
+    "DeepAI Generator": "Yntec/stable-diffusion-v1-5",
+    "Realistic Vision": "Yntec/stable-diffusion-v1-5",
+    "Artistic AI": "Yntec/stable-diffusion-v1-5"
 }
 
 def handle_input(prompt):