Update app.py

app.py

@@ -45,10 +45,11 @@ def sentiment_analysis(sentence, model_name):
     # for t in translated:
     # print( tokenizer.decode(t, skip_special_tokens=True) )
    
-    output = [sentence ['translation_text'] for sentence in tokenizer.decode(translated_tokens, skip_special_tokens=True)]
+    # output = [sentence ['translation_text'] for sentence in tokenizer.decode(translated_tokens, skip_special_tokens=True)]
+    for t in translated_tokens:
+        output.append(tokenizer.decode(t, skip_special_tokens=True)))
 
-    # nlp = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
-    # output = nlp(sentences)
+        
     
     # testdata = []
     # for i,sentence in enumerate(sentences):
@@ -84,7 +85,7 @@ def sentiment_analysis(sentence, model_name):
     #     logits = prompt_model(inputs)
     #     result.extend(torch.argmax(logits, dim=-1))
     # output = '\n'.join([classes[i] for i in result])
-    return output 
+    return str(output)