Update app.py
Browse files
app.py
CHANGED
@@ -28,6 +28,7 @@ try:
|
|
28 |
tokenizer = AutoTokenizer.from_pretrained(
|
29 |
MODEL_NAME,
|
30 |
cache_dir=str(cache_dir)
|
|
|
31 |
|
32 |
model = AutoModelForCausalLM.from_pretrained(
|
33 |
MODEL_NAME,
|
|
|
28 |
tokenizer = AutoTokenizer.from_pretrained(
|
29 |
MODEL_NAME,
|
30 |
cache_dir=str(cache_dir)
|
31 |
+
)
|
32 |
|
33 |
model = AutoModelForCausalLM.from_pretrained(
|
34 |
MODEL_NAME,
|