waqasali1707 commited on
Commit
4584bb0
·
verified ·
1 Parent(s): ee9c1e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -1,6 +1,13 @@
1
  import streamlit as st
2
  import torch
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
 
 
 
 
 
 
 
4
 
5
  # Update this path to your local path where the model is stored
6
  model_path = '/content/drive/MyDrive/bart-base'
 
1
  import streamlit as st
2
  import torch
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, GenerationConfig
4
+ from pydrive2.auth import GoogleAuth
5
+ from pydrive2.drive import GoogleDrive
6
+
7
+ # Authenticate and create the PyDrive client.
8
+ gauth = GoogleAuth()
9
+ gauth.LocalWebserverAuth() # Creates a local webserver and automatically handles authentication.
10
+ drive = GoogleDrive(gauth)
11
 
12
  # Update this path to your local path where the model is stored
13
  model_path = '/content/drive/MyDrive/bart-base'