Create diff.py

diff.py

@@ -0,0 +1,60 @@
+import gradio as gr
+class SendIt:
+    def __init__(self):
+        self.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",
+        ]
+        
+        self.models2=[
+            gr.Interface.load(f"models/{self.models[1]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[2]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[3]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[4]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[5]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[6]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[7]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[8]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[9]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[10]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[11]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[12]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[13]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[14]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[15]}",live=True,preprocess=True),   
+            gr.Interface.load(f"models/{self.models[16]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[17]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[18]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[19]}",live=True,preprocess=True),
+            gr.Interface.load(f"models/{self.models[20]}",live=True,preprocess=True),
+        ]
+
+    def send_it(self,inputs,model_choice, count = 1):
+        t=0
+        output_list=[]
+        while t<int(count):
+            proc1=self.models2[int(model_choice)]
+            output1=proc1(inputs)
+            output_list.append(output1)
+            inputs=f"{inputs} "
+            t+=1
+        return(output_list)