Update app.py

app.py

@@ -13,7 +13,7 @@ def generate(text):
     # 入力テキストをトークナイズ
     inputs = tokenizer.encode_plus(text, return_tensors='pt')
     # トークナイズされたテキストを使用して攻撃性を推定
-    outputs = np.array(model(inputs['input_ids'], inputs['attention_mask'])[:3])
+    outputs = model(inputs['input_ids'], inputs['attention_mask']).detach().numpy()[:3]
     # スコアを算出
     min = min(outputs)
     if min < 0:
@@ -35,4 +35,4 @@ iface = gr.Interface(
     title = "日本語のSNSコメントの攻撃性推定")
 
 # Gradio UIを起動
-iface.launch()
+iface.launch(share=True)