abdulllah01 commited on
Commit
594f70f
1 Parent(s): f8bc587

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -1,4 +1,12 @@
1
  from google.colab import drive
 
2
 
3
  # Mount Google Drive
4
- drive.mount('/content/drive')
 
 
 
 
 
 
 
 
1
  from google.colab import drive
2
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
  # Mount Google Drive
5
+ drive.mount('/content/drive')
6
+
7
+ # Path to the saved model on Google Drive
8
+ model_path = '/content/drive/My Drive/T5_samsum'
9
+
10
+ # Load the tokenizer and model from the specified path
11
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
12
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_path)