menikev commited on
Commit
46b4f79
·
verified ·
1 Parent(s): 38188b4

Update crypto_analysis_agents.py

Browse files
Files changed (1) hide show
  1. crypto_analysis_agents.py +5 -4
crypto_analysis_agents.py CHANGED
@@ -11,12 +11,13 @@ import os
11
  class CryptoAnalysisAgents:
12
  def __init__(self):
13
  self.llm = HuggingFaceHub(
14
- repo_id="distilbert-base-uncased",
15
- model_kwargs={"temperature": 0.7, "max_length": 512}
 
16
  )
17
- self.embeddings = HuggingFaceEmbeddings(model_name="distilbert-base-uncased")
18
  huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
19
-
20
  def market_analyst(self):
21
  return Agent(
22
  role='Crypto Market Analyst',
 
11
  class CryptoAnalysisAgents:
12
  def __init__(self):
13
  self.llm = HuggingFaceHub(
14
+ repo_id="facebook/opt-125m",
15
+ model_kwargs={"temperature": 0.7, "max_length": 512},
16
+ task="text-generation"
17
  )
18
+ self.embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
19
  huggingfacehub_api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
20
+
21
  def market_analyst(self):
22
  return Agent(
23
  role='Crypto Market Analyst',