Update src/prediction.py
Browse files- src/prediction.py +1 -1
src/prediction.py
CHANGED
|
@@ -2,7 +2,7 @@ from transformers import FlaxAutoModelForSeq2SeqLM
|
|
| 2 |
from transformers import AutoTokenizer
|
| 3 |
import textwrap
|
| 4 |
|
| 5 |
-
MODEL_NAME_OR_PATH = "flax-
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME_OR_PATH, use_fast=True)
|
| 7 |
model = FlaxAutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME_OR_PATH)
|
| 8 |
|
|
|
|
| 2 |
from transformers import AutoTokenizer
|
| 3 |
import textwrap
|
| 4 |
|
| 5 |
+
MODEL_NAME_OR_PATH = "jejun/flax-recipe-generator"
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME_OR_PATH, use_fast=True)
|
| 7 |
model = FlaxAutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME_OR_PATH)
|
| 8 |
|