Logeswaransr commited on
Commit
f74f4fd
1 Parent(s): 5870efb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ import streamlit as st
9
  import sentencepiece
10
  from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
11
 
12
- model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/results_T5").to("cpu")
13
- tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/results_T5")
14
 
15
  pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
16
 
 
9
  import sentencepiece
10
  from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
11
 
12
+ model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/T5_MineAI_Prototype").to("cpu")
13
+ tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/T5_MineAI_Prototype")
14
 
15
  pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
16