Update app.py

app.py

@@ -59,10 +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)
-    
-    return tokenizer.decode(outputs[0])
+    output = tokenizer.decode(outputs[0])
+    return ".".join(output.split(".")[:-1]) + "."
 
 
+output_text = gr.outputs.Textbox()
 
 
 if __name__ == '__main__':
@@ -81,7 +82,7 @@ app = gr.Interface(
 
     fn=generate,
 
-    inputs=[gr.inputs.Dropdown(model_name_list, label="Model Name"), 'text'], outputs=['label'], 
+    inputs=[gr.inputs.Dropdown(model_name_list, label="Model Name"), 'text'], output_text], 
 
     examples = [[MODEL_BUF["name"], text]],