Update app.py

app.py

@@ -3,14 +3,23 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import os
 
-# Retrieve the token and strip any whitespace or newline characters
+# Retrieve the token from environment variables
 api_token = os.getenv("HF_TOKEN").strip()
 
-# Load the Hugging Face model and tokenizer with the cleaned token
+# Model name
 model_name = "ContactDoctor/Bio-Medical-MultiModal-Llama-3-8B-V1"
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=api_token)
-model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, use_auth_token=api_token)
 
+# Load the Hugging Face model and tokenizer with required arguments
+tokenizer = AutoTokenizer.from_pretrained(
+    model_name, 
+    use_auth_token=api_token,  # Authenticate with Hugging Face token
+    trust_remote_code=True     # Allow custom code from the repository
+)
+model = AutoModelForCausalLM.from_pretrained(
+    model_name, 
+    use_auth_token=api_token,  # Authenticate with Hugging Face token
+    trust_remote_code=True     # Allow custom code from the repository
+)
 
 
 # Define the function to process user input