Spaces:

momoyukki
/

space-demo

Sleeping

momoyukki commited on May 3

Commit

ee723c8

•

1 Parent(s): b25ff40

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import nltk
 import math
 import torch
-# model_name = "fabiochiu/t5-base-medium-title-generation"
-model_name = "momoyukki/t5-base-news-title-generation_model"
 max_input_length = 512
 st.header("News Title Generation")

 import math
 import torch
+model_name = "fabiochiu/t5-base-medium-title-generation"
+#model_name = "momoyukki/t5-base-news-title-generation_model"
 max_input_length = 512
 st.header("News Title Generation")