MindVR commited on
Commit
c85c1b5
·
verified ·
1 Parent(s): f9d10a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ login(token=os.environ["HF_TOKEN"]) # Dùng biến môi trường để lấy t
5
  import torch
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import gradio as gr
8
-
9
  # Load model
10
  model_id = "MindVR/JohnTran_Fine-tune" # ⚠️ Đảm bảo đây là bản mới fine-tune không dùng 4bit
11
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ["HF_TOKEN"])
 
5
  import torch
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
  import gradio as gr
8
+
9
  # Load model
10
  model_id = "MindVR/JohnTran_Fine-tune" # ⚠️ Đảm bảo đây là bản mới fine-tune không dùng 4bit
11
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=os.environ["HF_TOKEN"])