Omnibus commited on
Commit
f6b485b
1 Parent(s): df37cde

Update dif.py

Browse files
Files changed (1) hide show
  1. dif.py +3 -5
dif.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  class SendIt:
3
- def send_it(self,inputs,model_choice=5, count = 1):
4
 
5
  models =[
6
  "",
@@ -52,13 +52,11 @@ class SendIt:
52
 
53
  ]
54
 
55
-
56
- if model_choice==None:
57
- model_choice=5
58
  t=0
59
  output_list=[]
60
  while t<int(count):
61
- proc1=models2[int(model_choice)]
62
  output1=proc1(inputs)
63
  output_list.append(output1)
64
  inputs=f"{inputs} "
 
1
  import gradio as gr
2
  class SendIt:
3
+ def __init__(self):
4
 
5
  models =[
6
  "",
 
52
 
53
  ]
54
 
55
+ def send_it(self,inputs,model_choice, count = 1):
 
 
56
  t=0
57
  output_list=[]
58
  while t<int(count):
59
+ proc1=self.models2[int(model_choice)]
60
  output1=proc1(inputs)
61
  output_list.append(output1)
62
  inputs=f"{inputs} "