Update app.py

app.py

@@ -15,8 +15,11 @@ torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer3 = PegasusTokenizer.from_pretrained(model_name)
 model3 = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)
 
+prev_context = "" # to store the previous context
 
 def qa_paraphrase(text_input, question):
+    global prev_context
+    text_input = prev_context + " " + text_input # combine with previous context
     prediction = classifier(
         context=text_input,
         question=question,
@@ -35,7 +38,8 @@ def qa_paraphrase(text_input, question):
     batch = tokenizer3([sentence],truncation=True,padding='longest',max_length=60, return_tensors="pt").to(torch_device)
     translated = model3.generate(**batch,max_length=60,num_beams=10, num_return_sequences=1, temperature=1.5)
     paraphrase = tokenizer3.batch_decode(translated, skip_special_tokens=True)[0]
-    return f"Answer: {answer}\nParaphrased Sentence: {paraphrase}"
+    prev_context = text_input # update previous context
+    return f"Q: {question}\nA: {answer}\nParaphrased Sentence: {paraphrase}"
 
 
 iface = gr.Interface(
@@ -49,4 +53,4 @@ iface = gr.Interface(
     description="Answer a given question from a given text and paraphrase the sentence containing the answer."
 )
 
-iface.launch()
+iface.launch()