Update app.py

app.py

@@ -6,9 +6,9 @@ access_token = os.environ["access_token"]
 
 title = "NX Block Trade: Multilingual Zero Shot News Classification"
 description = "powered by xlm-roberta-large-xnli made by @abhisheky127"
-classifier = pipeline("zero-shot-classification",
-                      model="joeddav/xlm-roberta-large-xnli",
-                     token=access_token)
+# classifier = pipeline("zero-shot-classification",
+#                       model="joeddav/xlm-roberta-large-xnli",
+#                      token=access_token)
 
 categories = ['sports',
  'IPL_2023',
@@ -84,42 +84,42 @@ test_examples = [
 ]
 
 
-# gr.Interface.load(model="models/joeddav/xlm-roberta-large-xnli", token=access_token,
-#                   title=title, description=description,
-#                  examples=examples_).launch()
+gr.Interface.load(model="models/joeddav/xlm-roberta-large-xnli", use_auth_token=access_token,
+                  title=title, description=description,
+                 examples=examples_).launch()
 
-#define a function to process your input and output
-def zero_shot(doc, candidates):
-    given_labels = candidates.split(",")
-    given_labels = list(map(str.strip, given_labels))
-    print(doc)
-    dictionary = classifier(doc, given_labels)
-    labels = dictionary['labels']
-    scores = dictionary['scores']
-    return dict(zip(labels, scores))
+# #define a function to process your input and output
+# def zero_shot(doc, candidates):
+#     given_labels = candidates.split(",")
+#     given_labels = list(map(str.strip, given_labels))
+#     print(doc)
+#     dictionary = classifier(doc, given_labels)
+#     labels = dictionary['labels']
+#     scores = dictionary['scores']
+#     return dict(zip(labels, scores))
 
 
 
-#create input and output objects
-#input object1
-input1 = gr.Textbox(label="Text")
+# #create input and output objects
+# #input object1
+# input1 = gr.Textbox(label="Text")
 
-#input object 2
-input2 = gr.Textbox(label="Labels")
+# #input object 2
+# input2 = gr.Textbox(label="Labels")
 
 
-#output object
-output = gr.Label(label="Output")
+# #output object
+# output = gr.Label(label="Output")
 
 
-#create interface
-gui = gr.Interface(title=title,
-                   description=description,
-                   fn=zero_shot,
-                   inputs=[input1, input2],
-                   outputs=[output],
-                   examples=test_examples
-                  )
+# #create interface
+# gui = gr.Interface(title=title,
+#                    description=description,
+#                    fn=zero_shot,
+#                    inputs=[input1, input2],
+#                    outputs=[output],
+#                    examples=test_examples
+#                   )
 
-#display the interface
-gui.launch(debug=True)
+# #display the interface
+# gui.launch(debug=True)