swcrazyfan commited on
Commit
9ee23d3
1 Parent(s): 9625079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import torch
2
  from transformers import (T5ForConditionalGeneration,T5Tokenizer)
3
  import gradio as gr
4
 
5
- best_model_path = "swcrazyfan/Dekingify-T5-Large"
6
  model = T5ForConditionalGeneration.from_pretrained(best_model_path)
7
- tokenizer = T5Tokenizer.from_pretrained("swcrazyfan/Dekingify-T5-Large")
8
 
9
  def tokenize_data(text, dekingify):
10
  # Tokenize the review body
 
2
  from transformers import (T5ForConditionalGeneration,T5Tokenizer)
3
  import gradio as gr
4
 
5
+ best_model_path = "swcrazyfan/Kingify-2Way-T5-Large-v1_1"
6
  model = T5ForConditionalGeneration.from_pretrained(best_model_path)
7
+ tokenizer = T5Tokenizer.from_pretrained("swcrazyfan/Kingify-2Way-T5-Large-v1_1")
8
 
9
  def tokenize_data(text, dekingify):
10
  # Tokenize the review body