Update app.py

app.py

@@ -1,10 +1,13 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
+
+# Login to Hugging Face with your token
 
 # Load the model and tokenizer
-model_name = "your-openbiollm-model-name"  # Replace with your model name
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model_name = "aaditya/Llama3-OpenBioLLM-8B"  # Replace with your model name
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
 
 # Define a function to generate text
 def generate_text(prompt):