Spaces:

javear
/

prediksi-icd

Runtime error

javear commited on Jul 11, 2023

Commit

ad0a18e

•

1 Parent(s): 26b4eea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,8 +15,8 @@ text = st.text_area('enter some text!')
 if text:
-    translated = translator.translate(text, lang_tgt='en')
-    encoded_input = tokenizer(translated, return_tensors='pt')
     output = model(**encoded_input)
     results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5]

 if text:
+    translated = translator.translate(text)
+    encoded_input = tokenizer(translated.text, return_tensors='pt')
     output = model(**encoded_input)
     results = output.logits.detach().cpu().numpy()[0].argsort()[::-1][:5]