Spaces:

OniXinO
/

SpaCy

Sleeping

OniXinO commited on 26 days ago

Commit

22986a6

1 Parent(s): 0cf9b7f

base

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torch
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/mt5-small", padding_side="left", use_fast=False)
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-small")
     return tokenizer, model
 st.title("Український Чат-бот")

 @st.cache_resource
 def load_model():
+    tokenizer = AutoTokenizer.from_pretrained("google/mt5-base", padding_side="left", use_fast=False)
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-base")
     return tokenizer, model
 st.title("Український Чат-бот")