jaydatech commited on
Commit
0f2cc4d
·
verified ·
1 Parent(s): 352bf61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -28,6 +28,7 @@ app.add_middleware(
28
 
29
  # Hugging Face model config
30
  REPO_NAME = "jaydatech/phi3-finetuned-project"
 
31
  HF_TOKEN = os.getenv("HF_TOKEN") # Load from Render environment variable
32
 
33
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -43,7 +44,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
43
  # )
44
 
45
  config = AutoConfig.from_pretrained(REPO_NAME, use_auth_token=HF_TOKEN)
46
- tokenizer = AutoTokenizer.from_pretrained(REPO_NAME, use_auth_token=HF_TOKEN)
47
 
48
  model = AutoModelForCausalLM.from_pretrained(
49
  REPO_NAME,
 
28
 
29
  # Hugging Face model config
30
  REPO_NAME = "jaydatech/phi3-finetuned-project"
31
+ BASE_MODEL = "microsoft/Phi-3-mini-4k-instruct"
32
  HF_TOKEN = os.getenv("HF_TOKEN") # Load from Render environment variable
33
 
34
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
44
  # )
45
 
46
  config = AutoConfig.from_pretrained(REPO_NAME, use_auth_token=HF_TOKEN)
47
+ tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, trust_remode_code=True, use_auth_token=HF_TOKEN)
48
 
49
  model = AutoModelForCausalLM.from_pretrained(
50
  REPO_NAME,