Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import T5ForConditionalGeneration, AutoTokenizer
 
 auth_token = os.environ.get("CLARIN_KNEXT")
 
-model_name = "clarin-knext/plt5-large-poquad-ext-qa-autotoken"
+model_name = "clarin-knext/plt5-large-poquad" # "clarin-knext/plt5-large-poquad-ext-qa-autotoken"
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=auth_token)
 model = T5ForConditionalGeneration.from_pretrained(model_name, use_auth_token=auth_token)
 
@@ -21,12 +21,12 @@ default_generate_kwargs = {
 # keywords_pipe = pipeline(model=model, tokenizer=tokenizer, **default_generate_kwargs)
 
 examples = [
-    ["extract question: Kiedy Arek ma spotkanie dzisiaj? context: Arek ma spotkanie przez cały dzeiń. Bardzo to lubi."]]
+    ["Jakie miasto jest stolicą Polski?", "Polska ma wiele wspaniałych miast, Wrocław, Poznań czy Gdańsk. Jednak stolicą jest Warszawa."]]
 
 
 
-def generate(context):
-    context = context + " </s>"
+def generate(question, context):
+    context = f"question: {question} context: {context} </s>"
     inputs = tokenizer(
         context, 
         max_length=512,
@@ -46,8 +46,8 @@ def generate(context):
 
 demo = gr.Interface(
     fn=generate,
-    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Keywords"),
+    inputs=[gr.Textbox(lines=1, label="Question"), gr.Textbox(lines=5, label="Context")],
+    outputs=gr.Textbox(label="Answer"),
     examples=examples,
 )