Hugging Face's logo

Spaces:
momoyukki
/
space-demo
Sleeping

App Files Community
momoyukki commited on
Commit
6d8829e
·
verified ·
1 Parent(s): e541492

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ import nltk
4
  import math
5
  import torch
6
 
7
- model_name = "momoyukki/t5-base-news-title-generation_model"
 
8
  max_input_length = 512
9
 
10
  st.header("Generate candidate titles for news")
 
4
  import math
5
  import torch
6
 
7
+ model_name = "fabiochiu/t5-base-medium-title-generation"
8
+ # model_name = "momoyukki/t5-base-news-title-generation_model"
9
  max_input_length = 512
10
 
11
  st.header("Generate candidate titles for news")