chages

app.py

@@ -106,33 +106,58 @@ with gr.Blocks() as app:
         return generate_with_embs(text_input, modified_output_embeddings, generator=g)
 
 
+    def generate_oil_painting(query):
+        return {
+            out1: func_generate(query, 0, 0)
+        }
+
+    def generate_low_poly_hd(query):
+        return {
+            out2: func_generate(query, 1, 30)
+        }
+
+    def generate_matrix_style(query):
+        return {
+            out3: func_generate(query, 2, 60)
+        }
+
+    def generate_dreamy_painting(query):
+        return {
+            out4: func_generate(query, 3, 90)
+        }
+
+    def generate_depth_map_style(query):
+        return {
+            out5: func_generate(query, 4, 120)
+        }
+
     submit1.click(
-        func_generate,
-        [search_text, 0, 0],
+        generate_oil_painting,
+        search_text,
         out1
     )
 
     submit2.click(
-        func_generate,
-        [search_text, 1, 30],
+        generate_low_poly_hd,
+        search_text,
         out2
     )
 
     submit3.click(
-        func_generate,
-        [search_text, 2, 60],
+        generate_matrix_style,
+        search_text,
         out3
     )
 
     submit4.click(
-        func_generate,
-        [search_text, 3, 90],
+        generate_dreamy_painting,
+        search_text,
         out4
     )
 
     submit5.click(
-        func_generate,
-        [search_text, 4, 120],
+        generate_depth_map_style,
+        search_text,
         out5
     )