Spaces:
Sleeping
Sleeping
Commit
•
f74f4fd
1
Parent(s):
5870efb
Update app.py
Browse files
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/
|
13 |
-
tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/
|
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 |
|