Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
96 |
#####
|
97 |
from huggingface_hub import InferenceClient
|
98 |
|
99 |
-
repo_id = "meta-llama/
|
100 |
|
101 |
llm = InferenceClient(model=repo_id, timeout=120)
|
102 |
|
@@ -114,8 +114,8 @@ def download_nltk_resources():
|
|
114 |
|
115 |
download_nltk_resources()
|
116 |
|
117 |
-
nltk.download('punkt')
|
118 |
-
|
119 |
FILES_DIR = './files'
|
120 |
|
121 |
|
|
|
96 |
#####
|
97 |
from huggingface_hub import InferenceClient
|
98 |
|
99 |
+
repo_id = "meta-llama/Llama-3.2-1B-Instruct"
|
100 |
|
101 |
llm = InferenceClient(model=repo_id, timeout=120)
|
102 |
|
|
|
114 |
|
115 |
download_nltk_resources()
|
116 |
|
117 |
+
#nltk.download('punkt')
|
118 |
+
print("------ Strinage ------ ")
|
119 |
FILES_DIR = './files'
|
120 |
|
121 |
|