ChienChung commited on
Commit
40f94cb
·
verified ·
1 Parent(s): d8607c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import re
8
  import requests
9
  import transformers
10
  import chardet
11
- import deepeval
12
  import difflib
13
  from transformers import AutoModelForCausalLM, AutoTokenizer
14
  from transformers.models.llama.configuration_llama import LlamaConfig
@@ -149,7 +149,7 @@ print(f"Using device => {device}")
149
 
150
  hf_token = os.environ.get("HF_TOKEN")
151
  openai_api_key = os.environ.get("OPENAI_API_KEY")
152
- model_id = "ChienChung/my-llama-1b"
153
 
154
  config_path = hf_hub_download(
155
  repo_id=model_id,
 
8
  import requests
9
  import transformers
10
  import chardet
11
+ #import deepeval
12
  import difflib
13
  from transformers import AutoModelForCausalLM, AutoTokenizer
14
  from transformers.models.llama.configuration_llama import LlamaConfig
 
149
 
150
  hf_token = os.environ.get("HF_TOKEN")
151
  openai_api_key = os.environ.get("OPENAI_API_KEY")
152
+ model_id = "meta-llama/Llama-3.2-1B-Instruct"
153
 
154
  config_path = hf_hub_download(
155
  repo_id=model_id,