Update app.py

app.py

@@ -28,24 +28,23 @@ def preprocess(temp):
     return temp
 
 def classify_text(inp):
-    print(inp)
+    # print(inp)
     inp = preprocess(inp)
-    print(inp)
+    # print(inp)
     inp = tokenizer(inp, padding=True, max_length=512, truncation=True, return_tensors="pt")
-    print(inp)
+    # print(inp)
     with torch.no_grad():
         logits = model(**inp).logits
-    print(logits)
-    print(logits.shape)
+    # print(logits)
+    # print(logits.shape)
     # logits = logits.argsort().squeeze(0)
     logits = torch.nn.Softmax(dim=1)(logits).squeeze(0)
-    print(logits)
+    # print(logits)
     return {labels[i]: float(logits[i].item()) for i in range(len(labels))}
     # return {logits.argmax().item() : labels[logits.argmax().item()]}
 
 input = '明天又是好日子\n千金的光阴不能等\n明天又是好日子\n赶上了盛世咱享太平\n今天是个好日子\n心想的事儿都能成\n明天又是好日子\n千金的光阴不能等\n今天明天都是好日子\n赶上了盛世咱享太平'
 
-
 gr.Interface(
     classify_text,
     # gr.inputs.Image(),
@@ -54,4 +53,4 @@ gr.Interface(
     # inputs='image',
     # outputs='label',
     # examples=[["images/cheetah1.jpg"], ["images/lion.jpg"]],
-).launch(debug=True)
+).launch(debug=True,share=True)