Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login
 import torch
 import re
 import os
@@ -9,6 +10,7 @@ model_name = "google/gemma-2b"
 peft_model = "kazuma313/gemma-dokter-ft"
 device_map = "auto"
 hf_token = os.getenv('hftoken')
+login(token=hf_token)
 
 config = PeftConfig.from_pretrained(peft_model)