Update crypto_analysis_agents.py

crypto_analysis_agents.py

@@ -10,16 +10,30 @@ class CryptoAnalysisAgents:
         model_name = "facebook/opt-350m"  # A smaller, open-source model
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(model_name)
+
+        # Create a custom generation config
+        generation_config = GenerationConfig(
+        max_new_tokens=512,
+        do_sample=True,
+        temperature=0.7,
+        top_p=0.95,
+        max_length=2048  # Set a large value for max_length
+        )
+
+        # Update the model' config
+        model.config.max_length = 2048
         
         pipe = pipeline(
             "text-generation",
             model=model,
             tokenizer=tokenizer,
+            generation_config=generation_config,
             max_new_tokens=512,
             do_sample=True,
             temperature=0.7,
             top_p=0.95,
             repetition_penalty=1.1
+            
         )
         
         self.llm = HuggingFacePipeline(pipeline=pipe)