Akshat-1812 commited on
Commit
710bac2
·
verified ·
1 Parent(s): 86d6fdc

trying to import model from Google drive using gdown

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -5,10 +5,14 @@ import tensorflow as tf
5
  import tensorflow_hub as hub
6
  from tensorflow.keras.layers import TextVectorization
7
  import gradio as gr
 
8
 
9
 
 
 
 
10
  # Load model
11
- model_path = "/content/drive/MyDrive/tensorflow_course/skim_lit/skimlit_tribrid_model"
12
  loaded_model = tf.keras.models.load_model(model_path)
13
 
14
  # create nlp english lang. pipline
 
5
  import tensorflow_hub as hub
6
  from tensorflow.keras.layers import TextVectorization
7
  import gradio as gr
8
+ import gdown
9
 
10
 
11
+ url = 'https://drive.google.com/drive/folders/1ATO7lCqHr3ZCkICAs2ensV4YBe_ORhhE?usp=drive_link'
12
+ gdown.download_folder(url)
13
+
14
  # Load model
15
+ model_path = "skimlit_tribrid_model_100"
16
  loaded_model = tf.keras.models.load_model(model_path)
17
 
18
  # create nlp english lang. pipline