Update app.py

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
+import os
+from huggingface_hub import login
+
+login(os.environ["HF_TOKEN"])
+
+
 # Load MedGemma
 MODEL_NAME = "google/medgemma-4b-it"  # choose the model size you want. here
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)