Futuresony commited on
Commit
f65fcfc
·
verified ·
1 Parent(s): 56d52d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -3,7 +3,11 @@ import faiss
3
  import numpy as np
4
  import torch
5
  import gradio as gr
6
- from transformers import AutoModelForCausalLM, GemmaTokenizer
 
 
 
 
7
  from huggingface_hub import hf_hub_download
8
 
9
  # 🔹 Hugging Face Credentials
 
3
  import numpy as np
4
  import torch
5
  import gradio as gr
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer
7
+
8
+ # ✅ Fix: Use AutoTokenizer instead of GemmaTokenizer
9
+ tokenizer = AutoTokenizer.from_pretrained(HF_REPO, token=HF_TOKEN)
10
+
11
  from huggingface_hub import hf_hub_download
12
 
13
  # 🔹 Hugging Face Credentials