Wootang01 commited on
Commit
7d77ba4
·
1 Parent(s): 9e38a61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- 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)
 
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)