Update app.py

app.py

@@ -40,7 +40,7 @@ def predict(query, context):
     context = context.decode()
 
     #Custom1
-    cust_model_name = "anshoomehra/question-answering-roberta-base-s"
+    cust_model_name = "consciousAI/question-answering-roberta-base-s"
     cust_question_answerer = pipeline('question-answering', model=cust_model_name, tokenizer=cust_model_name, device=device)
     
     cust_output = cust_question_answerer(question=query, context=context)
@@ -54,7 +54,7 @@ def predict(query, context):
         cust_answer_sentence = "Failed matching sentence (answer may be split in multiple sentences)"
         
     #Custom2
-    cust_answer_2 = _generate(query, context, model="anshoomehra/question-answering-generative-t5-v1-base-s-q-c", device=device)
+    cust_answer_2 = _generate(query, context, model="consciousAI/question-answering-generative-t5-v1-base-s-q-c", device=device)
     cust_answer_sentence_2 = [_sent for _sent in sent_tokenize(context) if cust_answer_2 in _sent]
     if len(cust_answer_sentence_2) > 0:
         cust_answer_sentence_2 = cust_answer_sentence_2[0]