Update dif.py

dif.py

@@ -1,66 +1,66 @@
 import gradio as gr
-
-models =[
-    "",
-    "CompVis/stable-diffusion-v1-4",
-    "runwayml/stable-diffusion-v1-5",
-    "prompthero/openjourney",
-    "stabilityai/stable-diffusion-2-1",
-    "stabilityai/stable-diffusion-2-1-base",
-    "SG161222/Realistic_Vision_V1.4",
-    "Linaqruf/anything-v3.0",
-    "eimiss/EimisAnimeDiffusion_1.0v",
-    "nitrosocke/Nitro-Diffusion",
-    "wavymulder/portraitplus",
-    "22h/vintedois-diffusion-v0-1",
-    "dreamlike-art/dreamlike-photoreal-2.0",
-    "dreamlike-art/dreamlike-diffusion-1.0",
-    "wavymulder/Analog-Diffusion",
-    "nitrosocke/redshift-diffusion",
-    "claudfuen/photorealistic-fuen-v1",
-    "prompthero/openjourney-v2",
-    "johnslegers/epic-diffusion",
-    "nitrosocke/Arcane-Diffusion",
-    "darkstorm2150/Protogen_x5.8_Official_Release",
-
-]
-current_model = models[5]
-
-models2=[
-    gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[4]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[5]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[6]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[7]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[8]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[9]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[10]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[11]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[12]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[13]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[14]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[15]}",live=True,preprocess=True),   
-    gr.Interface.load(f"models/{models[16]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[17]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[18]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[19]}",live=True,preprocess=True),
-    gr.Interface.load(f"models/{models[20]}",live=True,preprocess=True),
-      
-
-]
-
-
-def send_it(inputs,model_choice=5, count = 1):
-    if model_choice==None:
-        model_choice=5
-    t=0
-    output_list=[]
-    while t<int(count):
-        proc1=models2[int(model_choice)]
-        output1=proc1(inputs)
-        output_list.append(output1)
-        inputs=f"{inputs} "
-        t+=1
-    return(output_list)
+class SendIt:
+    
+    models =[
+        "",
+        "CompVis/stable-diffusion-v1-4",
+        "runwayml/stable-diffusion-v1-5",
+        "prompthero/openjourney",
+        "stabilityai/stable-diffusion-2-1",
+        "stabilityai/stable-diffusion-2-1-base",
+        "SG161222/Realistic_Vision_V1.4",
+        "Linaqruf/anything-v3.0",
+        "eimiss/EimisAnimeDiffusion_1.0v",
+        "nitrosocke/Nitro-Diffusion",
+        "wavymulder/portraitplus",
+        "22h/vintedois-diffusion-v0-1",
+        "dreamlike-art/dreamlike-photoreal-2.0",
+        "dreamlike-art/dreamlike-diffusion-1.0",
+        "wavymulder/Analog-Diffusion",
+        "nitrosocke/redshift-diffusion",
+        "claudfuen/photorealistic-fuen-v1",
+        "prompthero/openjourney-v2",
+        "johnslegers/epic-diffusion",
+        "nitrosocke/Arcane-Diffusion",
+        "darkstorm2150/Protogen_x5.8_Official_Release",
+    
+    ]
+    
+    models2=[
+        gr.Interface.load(f"models/{models[1]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[2]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[3]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[4]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[5]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[6]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[7]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[8]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[9]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[10]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[11]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[12]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[13]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[14]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[15]}",live=True,preprocess=True),   
+        gr.Interface.load(f"models/{models[16]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[17]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[18]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[19]}",live=True,preprocess=True),
+        gr.Interface.load(f"models/{models[20]}",live=True,preprocess=True),
+          
+    
+    ]
+    
+    
+    def send_it(self,inputs,model_choice=5, count = 1):
+        if model_choice==None:
+            model_choice=5
+        t=0
+        output_list=[]
+        while t<int(count):
+            proc1=models2[int(model_choice)]
+            output1=proc1(inputs)
+            output_list.append(output1)
+            inputs=f"{inputs} "
+            t+=1
+        return(output_list)