Update app.py

app.py

@@ -1,5 +1,3 @@
-from diffusers import StableDiffusionPipeline
-import torch
 import gradio
 import accelerate
 
@@ -31,15 +29,16 @@ def process1(prompt):
     for i in range(len(models)):
         if prompt.find(models[i].prefix)!=-1:
             modelSelected=models[i].path
-            gradio.Interface.load(modelSelected)
+            m=gradio.Interface.load(modelSelected)
     if (modelSelected==''):
         modelSelected = "models/stabilityai/stable-diffusion-2-1"
-        gradio.Interface.load(modelSelected)
-    image_return = modelSelected(prompt)
+        m=gradio.Interface.load(modelSelected)
+    print(modelSelected)
+    image_return = m(prompt)
     return image_return
 
 sandbox = gradio.Interface(fn=process1, 
                         inputs=[gradio.Textbox(label="Enter Prompt:")],
                         outputs=[gradio.Image(label="Produced Image")], 
                         title='AlStable Text to Image')
-sandbox.queue(concurrency_count=20).launch()
+sandbox.queue(concurrency_count=20).launch()