Spaces:
Sleeping
Sleeping
grpathak22
commited on
Commit
•
2aa73f0
1
Parent(s):
671b9b5
Update modelrun.py
Browse files- modelrun.py +9 -0
modelrun.py
CHANGED
@@ -1,3 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
tokenizer = T5Tokenizer.from_pretrained('google/mt5-base')
|
2 |
model = MT5ForConditionalGeneration.from_pretrained("google/mt5-base")
|
3 |
#st.write(model)
|
|
|
1 |
+
from transformers import AutoTokenizer, MT5ForConditionalGeneration
|
2 |
+
from transformers import T5Tokenizer
|
3 |
+
import streamlit as st
|
4 |
+
import pandas as pd
|
5 |
+
from datasets import Dataset
|
6 |
+
import torch
|
7 |
+
from datasets import Dataset, DatasetDict
|
8 |
+
from transformers import Trainer, TrainingArguments
|
9 |
+
|
10 |
tokenizer = T5Tokenizer.from_pretrained('google/mt5-base')
|
11 |
model = MT5ForConditionalGeneration.from_pretrained("google/mt5-base")
|
12 |
#st.write(model)
|