Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -8,37 +8,31 @@ import os
 
 class CryptoAnalysisAgents:
     def __init__(self):
-        # Load the GPT-Neo 2.7B model and tokenizer
-        model_name = "EleutherAI/gpt-neo-2.7B"
+        # Load a smaller, open-source model
+        model_name = "facebook/opt-350m"  # You can change this to another open-source model
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(model_name)
-
-    def generate_report(self, input_text):
-        truncated_input = self._truncate_input(input_text)
-        result = self.pipe(truncated_input, max_length=1024, max_new_tokens=512)
-        return result[0]['generated_text']
-
-        # Create the text-generation pipeline
+        
+        # Create a text-generation pipeline
         pipe = pipeline(
             "text-generation",
             model=model,
             tokenizer=tokenizer,
-            max_length=1024,  # Use max_length instead of max_new_tokens
+            max_new_tokens=512,
             do_sample=True,
             temperature=0.7,
             top_p=0.95,
             top_k=50,
-            repetition_penalty=1.1,
-            pad_token_id=tokenizer.eos_token_id
+            repetition_penalty=1.1
         )
-
-        # Wrap the pipeline in a HuggingFacePipeline
+        
+        # Create a LangChain wrapper around the Hugging Face pipeline
         self.llm = HuggingFacePipeline(pipeline=pipe)
 
     def market_analyst(self):
         return Agent(
             role='Crypto Market Analyst',
-            goal="Provide in-depth industry, market analysis and insights for cryptocurrencies. You also provide insight on on our geo-politics and economic policies impact on crypto",
+            goal="Provide in-depth industry, market analysis and insights for cryptocurrencies. You also provide insight on our geo-politics and economic policies impact on crypto",
             backstory="""You are a seasoned crypto market analyst with years of experience in blockchain technology and cryptocurrency markets. Your expertise helps clients navigate the volatile crypto landscape.""",
             verbose=True,
             llm=self.llm,