Update app.py

app.py

@@ -59,11 +59,8 @@ def generate(model_name, text):
     model.eval()
     input_ids = tokenizer.encode("AFA:{}".format(text), return_tensors="pt")
     outputs = model.generate(input_ids, max_length=200, num_beams=2, repetition_penalty=2.5, top_k=50, top_p=0.98, length_penalty=1.0, early_stopping=True)
-    result = {}
     
-    for idx, label in enumerate(output[0].detach().numpy()):
-        result[config.id2label[idx]] = float(label)
-    return result
+    return tokenizer.decode(outputs[0])