Update app.py

app.py

@@ -59,8 +59,11 @@ 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 = {}
     
-    return tokenizer.decode(outputs[0])
+    for idx, label in enumerate(output[0].detach().numpy()):
+        result[config.id2label[idx]] = float(label)
+    return result