Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -7,26 +7,22 @@ from sentiment_tools import SentimentTools
 
 class CryptoAnalysisAgents:
     def __init__(self):
-        model_name = "facebook/opt-350m"  # Or another suitable model
+        model_name = "facebook/opt-350m"  # A smaller, open-source model
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(model_name)
         
-        def custom_text_generation(prompt, max_new_tokens=512):
-            inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=max_new_tokens,
-                do_sample=True,
-                temperature=0.7,
-                top_p=0.95,
-                top_k=50,
-                repetition_penalty=1.1,
-                pad_token_id=tokenizer.eos_token_id,
-                eos_token_id=tokenizer.eos_token_id,
-            )
-            return tokenizer.decode(outputs[0], skip_special_tokens=True)
+        pipe = pipeline(
+            "text-generation",
+            model=model,
+            tokenizer=tokenizer,
+            max_new_tokens=512,
+            do_sample=True,
+            temperature=0.7,
+            top_p=0.95,
+            repetition_penalty=1.1
+        )
         
-        self.llm = HuggingFacePipeline(pipeline=custom_text_generation)
+        self.llm = HuggingFacePipeline(pipeline=pipe)
 
     def market_analyst(self):
         return Agent(