Upload app.py

app.py

@@ -12,16 +12,14 @@ from keras.models import load_model
 
 
 def text_clf(text):
-    # os.environ['CUDA_VISIBLE_DEVICES'] = '1'
-    # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     vocab_file = "vocab.txt"  # 词汇表
     tokenizer = BertTokenizer(vocab_file)
     # 加载模型
     config = BertConfig.from_pretrained("nanaaaa/emotion_chinese_english")
     model = XLMRobertaForSequenceClassification.from_pretrained("nanaaaa/emotion_chinese_english", config=config)
-    # model.to(device)
+
     inputs = tokenizer(text, return_tensors="pt")
-    # inputs.to(device)
+
     # 模型推断
     outputs = model(**inputs)
     probs = torch.nn.functional.softmax(outputs.logits, dim=1)
@@ -144,4 +142,4 @@ with gr.Blocks() as demo:
     audio_button.click(fn=audio_clf, inputs=audio, outputs=audio_output)
     cir_button.click(fn=cir_clf, inputs=[cir_l, cir_r], outputs=cir_output)
 
-demo.launch(share=True)
+demo.launch()