Spaces:

ethanrom
/

chat2

Runtime error

ethanrom commited on Mar 3, 2023

Commit

d90d622

1 Parent(s): 0e55ca4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ qa_classifier = pipeline(
 paraphrase_model_name = 'tuner007/pegasus_paraphrase'
 paraphrase_tokenizer = PegasusTokenizer.from_pretrained(paraphrase_model_name)
 paraphrase_model = PegasusForConditionalGeneration.from_pretrained(paraphrase_model_name)
 def find_answer_and_paraphrase(text_input, question):
     # Find the answer

 paraphrase_model_name = 'tuner007/pegasus_paraphrase'
 paraphrase_tokenizer = PegasusTokenizer.from_pretrained(paraphrase_model_name)
 paraphrase_model = PegasusForConditionalGeneration.from_pretrained(paraphrase_model_name)
+torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 def find_answer_and_paraphrase(text_input, question):
     # Find the answer