Update app.py

app.py

@@ -3,7 +3,16 @@ import gradio as gr
 
 models = [
     {"name": "Stable Diffusion 2", "url": "stabilityai/stable-diffusion-2-1"},
-]    
+    {"name": "stability AI", "url": "stabilityai/stable-diffusion-2-1-base"},
+    {"name": "XL-Refiner-1.0", "url": "stabilityai/stable-diffusion-xl-refiner-1.0"},
+    {"name": "Future Diffusion", "url": "nitrosocke/Future-Diffusion"},
+    {"name": "JWST Deep Space Diffusion", "url": "dallinmackay/JWST-Deep-Space-diffusion"},
+    {"name": "Robo Diffusion 3 Base", "url": "nousr/robo-diffusion-2-base"},
+    {"name": "Robo Diffusion", "url": "nousr/robo-diffusion"},
+    {"name": "Tron Legacy Diffusion", "url": "dallinmackay/Tron-Legacy-diffusion"},
+]   
+
+text_gen = gr.Interface.load("spaces/daspartho/prompt-extend")
 
 current_model = models[0]
 
@@ -13,6 +22,9 @@ for model in models:
     loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
     models2.append(loaded_model)
 
+def text_it(inputs, text_gen=text_gen):
+    return text_gen(inputs)
+
 def flip_text(x):
     return x[::-1]