grpathak22 commited on
Commit
2aa73f0
1 Parent(s): 671b9b5

Update modelrun.py

Browse files
Files changed (1) hide show
  1. 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)