Update app.py

app.py

@@ -8,7 +8,7 @@ import requests
 import gradio as gr
 import numpy as np
 from lxml.html import fromstring
-from transformers import pipeline
+#from transformers import pipeline
 from torch import multiprocessing as mp
 #from pathos.multiprocessing import ProcessPool as Pool
 from pathos.threading import ThreadPool as Pool
@@ -123,7 +123,7 @@ def Piper2(pipe,img,posi,neg):
         return None
 
 @spaces.GPU(duration=55)
-def tok(txt):
+def tok(pipe,txt):
     toks = pipe.tokenizer(txt)['input_ids']
     print(toks)
     return toks
@@ -194,14 +194,14 @@ function custom(){
 }
 """
 
-if __name__ == "__main__":
-    
+def main():
+
     pp1=pipe_t2i()
     pp2=pipe_i2i()
     
     result = []
     
-    def main(p1,p2,*result):
+    def run(p1,p2,*result):
         p1_en = translate(p1,"english")
         p2_en = translate(p2,"english")
         p = {"a":p1_en,"b":p2_en}
@@ -240,5 +240,8 @@ if __name__ == "__main__":
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
                 result.append(gr.Image(interactive=False,elem_classes="image-container", label="Result", show_label=False, type='filepath', show_share_button=False))
 
-        run_button.click(fn=main,inputs=[prompt,prompt2,*result],outputs=result)
-        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
+        run_button.click(fn=run,inputs=[prompt,prompt2,*result],outputs=result)
+        demo.queue().launch(server_port=int(os.getenv("CUSTOM_PORT")))
+
+if __name__ == "__main__":
+    main()