AliArshad commited on
Commit
1591131
1 Parent(s): eece5d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -2,6 +2,11 @@ import torch
2
  from transformers import XLNetTokenizer, XLNetForSequenceClassification
3
  import gradio as gr
4
 
 
 
 
 
 
5
  # Path to the saved model
6
  model_path = '/content/drive/My Drive/XLNet_model_project_Core.pt' # Update with your model path
7
 
 
2
  from transformers import XLNetTokenizer, XLNetForSequenceClassification
3
  import gradio as gr
4
 
5
+
6
+ from google.colab import drive
7
+
8
+ # Mount Google Drive
9
+ drive.mount('/content/drive')
10
  # Path to the saved model
11
  model_path = '/content/drive/My Drive/XLNet_model_project_Core.pt' # Update with your model path
12