Update app.py
Browse files
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 = "
|
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,
|