Update src/RAGSample.py

src/RAGSample.py

@@ -348,29 +348,6 @@ def setup_retriever(use_kaggle_data: bool = False, kaggle_dataset: Optional[str]
 #     # Create a chain combining the prompt template and LLM
 #     return prompt | llm | StrOutputParser()
 
-def initialize_biogpt():
-    try:
-        hf_pipeline = pipeline(
-            "text-generation",
-            model="microsoft/BioGPT",
-            tokenizer="microsoft/BioGPT",
-            max_new_tokens=100,
-            # Remove or comment out max_length to avoid truncation
-            # max_length=1024,
-            temperature=0.2,
-            device_map="auto",
-            torch_dtype=torch.float16,
-            return_full_text=False,
-            truncation=False,           # Explicitly disable truncation
-            do_sample=True,
-            pad_token_id=1,
-            eos_token_id=2,
-            )
-        print("BioGPT loaded successfully!")
-        return hf_pipeline
-    except Exception as e:
-        print(f"Error loading BioGPT: {e}")
-        return None
 
 def setup_rag_chain() -> Runnable:
     """Sets up the RAG chain with a prompt template and an LLM."""
@@ -392,8 +369,6 @@ Answer:
     )
     
     # Initialize a local Hugging Face model
-    hf_pipeline = pipeline(
-            # Initialize a local Hugging Face model
     hf_pipeline = pipeline(
         "text-generation",
         model="microsoft/BioGPT",