Update src/RAGSample.py

src/RAGSample.py

@@ -351,19 +351,14 @@ def initialize_biogpt():
     try:
         hf_pipeline = pipeline(
             "text-generation",
-            model="microsoft/BioGPT",
-            tokenizer="microsoft/BioGPT",
+            model="microsoft/DialoGPT-medium",  # No extra dependencies needed
             max_new_tokens=150,
             temperature=0.3,
             device_map="auto",
-            torch_dtype=torch.float16,
             return_full_text=False,
             truncation=True,
             do_sample=True,
-            top_p=0.9,
-            repetition_penalty=1.1,
-            pad_token_id=1,
-            eos_token_id=2,
+            pad_token_id=50256,  # DialoGPT pad token
         )
         print("BioGPT loaded successfully!")
         return hf_pipeline