Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -39,9 +39,9 @@ def diff_strings(a, b):
|
|
39 |
return result
|
40 |
|
41 |
@st.cache(suppress_st_warning=True, allow_output_mutation=True)
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
|
46 |
happy_tt = get_happy_text(checkpoint)
|
47 |
args = TTSettings(num_beams=5, min_length=1)
|
|
|
39 |
return result
|
40 |
|
41 |
@st.cache(suppress_st_warning=True, allow_output_mutation=True)
|
42 |
+
def get_happy_text(model_name):
|
43 |
+
st.write(f"Loading the HappyTextToText model {model_name}, please wait...")
|
44 |
+
return HappyTextToText("T5", model_name)
|
45 |
|
46 |
happy_tt = get_happy_text(checkpoint)
|
47 |
args = TTSettings(num_beams=5, min_length=1)
|