Hugging Face's logo

roneliav
/
QA_discourse_v2

Text2Text Generation
Transformers
PyTorch
t5
Inference Endpoints
text-generation-inference
Model card Files Community
1
roneliav commited on
Commit
62bfe7a
•
1 Parent(s): 8a6ef25

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -65,7 +65,7 @@ class QADiscourse_Pipeline(Text2TextGenerationPipeline):
65
 
66
 
67
  if __name__ == "__main__":
68
- pipe = QADiscourse_Pipeline("RonEliav/QA_discourse")
69
  res1 = pipe("I don't like chocolate, but I like cookies.")
70
  res2 = pipe(["I don't like chocolate, but I like cookies.",
71
  "I dived in the sea easily"], num_beams=10)
 
65
 
66
 
67
  if __name__ == "__main__":
68
+ pipe = QADiscourse_Pipeline("RonEliav/QA_discourse_v2")
69
  res1 = pipe("I don't like chocolate, but I like cookies.")
70
  res2 = pipe(["I don't like chocolate, but I like cookies.",
71
  "I dived in the sea easily"], num_beams=10)