Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,9 @@ import time
|
|
7 |
from queue import Queue
|
8 |
from threading import Thread
|
9 |
|
|
|
|
|
|
|
10 |
API_URL = "models/facebook/mbart-large-50-many-to-one-mmt"
|
11 |
API_TOKEN = os.environ.get("HF_READ_TOKEN")
|
12 |
|
@@ -80,6 +83,8 @@ import random
|
|
80 |
import time
|
81 |
|
82 |
# ... (existing code)
|
|
|
|
|
83 |
|
84 |
def get_prompts(prompt_text):
|
85 |
if not prompt_text:
|
|
|
7 |
from queue import Queue
|
8 |
from threading import Thread
|
9 |
|
10 |
+
|
11 |
+
|
12 |
+
|
13 |
API_URL = "models/facebook/mbart-large-50-many-to-one-mmt"
|
14 |
API_TOKEN = os.environ.get("HF_READ_TOKEN")
|
15 |
|
|
|
83 |
import time
|
84 |
|
85 |
# ... (existing code)
|
86 |
+
model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
|
87 |
+
tokenizer = MBart50TokenizerFast
|
88 |
|
89 |
def get_prompts(prompt_text):
|
90 |
if not prompt_text:
|