minor

paper_chat_tab.py

@@ -167,7 +167,7 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
         # Load the tokenizer from Hugging Face
         # tokenizer = AutoTokenizer.from_pretrained(model_name)
         tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct",
-                                                  token=os.environ.get("HF_TOKEN"))
+                                                  token=os.getenv("HF_TOKEN"))
         tokenizer_cache[model_name] = tokenizer
     else:
         tokenizer = tokenizer_cache[model_name]
@@ -241,7 +241,7 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
         final_messages.extend(messages)
 
         # Use the Hugging Face token if provided
-        api_key = hf_token_value or os.environ.get("SAMBANOVA_API_KEY")
+        api_key = hf_token_value or os.getenv("SAMBANOVA_API_KEY")
         if not api_key:
             raise ValueError("API token is not provided.")
 

requirements.txt

@@ -4,7 +4,7 @@ datasets
 scholarly
 arxiv
 PyPDF2
-transformers
+transformers=4.46.3
 beautifulsoup4
 openai
 # Set the primary index URL to PyTorch's CPU wheels