Spaces:
Runtime error
Runtime error
Commit
·
888419c
1
Parent(s):
42dfeae
Remove persistent caching
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ code_to_name = {
|
|
43 |
"es": "Spanish",
|
44 |
}
|
45 |
|
46 |
-
@st.cache
|
47 |
def generate_sequence(pixel_values, lang_code, num_beams, temperature, top_p):
|
48 |
lang_code = language_mapping[lang_code]
|
49 |
output_ids = model.generate(input_ids=pixel_values, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code], max_length=64, num_beams=num_beams, temperature=temperature, top_p = top_p)
|
|
|
43 |
"es": "Spanish",
|
44 |
}
|
45 |
|
46 |
+
@st.cache
|
47 |
def generate_sequence(pixel_values, lang_code, num_beams, temperature, top_p):
|
48 |
lang_code = language_mapping[lang_code]
|
49 |
output_ids = model.generate(input_ids=pixel_values, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code], max_length=64, num_beams=num_beams, temperature=temperature, top_p = top_p)
|