VPCSinfo commited on
Commit
74d45f5
·
1 Parent(s): 8895970

[ADD] added fixes for tool model.

Browse files
Files changed (1) hide show
  1. tool.py +1 -1
tool.py CHANGED
@@ -17,7 +17,7 @@ class TranscriptSummarizer(Tool):
17
 
18
  def __init__(self, *args, **kwargs):
19
  super().__init__(*args, **kwargs)
20
- self.summarizer = pipeline("summarization", model="facebook/usin")
21
  self.api_url = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
22
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
23
 
 
17
 
18
  def __init__(self, *args, **kwargs):
19
  super().__init__(*args, **kwargs)
20
+ self.summarizer = pipeline("summarization", model="google/pegasus-xsum")
21
  self.api_url = "https://api-inference.huggingface.co/models/ZB-Tech/Text-to-Image"
22
  self.headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
23