Upload app.py

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
-from simpletransformers.question_answering import QuestionAnsweringModel
 
 # Load model from training checkpoint
+from simpletransformers.question_answering import QuestionAnsweringModel, QuestionAnsweringArgs
+
 model = QuestionAnsweringModel("bert", "./bert_outputs/bert/best_model", use_cuda=False)
 
 # Função para realizar predições
@@ -14,10 +15,10 @@ def perform_prediction(context, question):
 iface = gr.Interface(
     fn=perform_prediction,
     inputs=[
-        gr.Textbox(placeholder="Insert context here...", label="Contexto"),
-        gr.Textbox(placeholder="Ask a question about the context...", label="Pergunta")
+        gr.Textbox(placeholder="Insira o contexto aqui...", label="Contexto"),
+        gr.Textbox(placeholder="Faça uma pergunta sobre o contexto...", label="Pergunta")
     ],
-    outputs=gr.Textbox(label="Response"),
+    outputs=gr.Textbox(label="Resposta"),
     live=True,
     theme="compact",
 )