Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ def get_model(model_name, model_path):
|
|
14 |
return model, tokenizer
|
15 |
|
16 |
|
17 |
-
|
18 |
def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_length=300):
|
19 |
input_ids = tokenizer.encode(text, return_tensors="pt")
|
20 |
with torch.no_grad():
|
|
|
14 |
return model, tokenizer
|
15 |
|
16 |
|
17 |
+
#@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
|
18 |
def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_length=300):
|
19 |
input_ids = tokenizer.encode(text, return_tensors="pt")
|
20 |
with torch.no_grad():
|