Winnie-Kay commited on
Commit
229809b
·
1 Parent(s): 5ae9a2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -3,6 +3,14 @@ os.system('pip install torch')
3
  os.system('pip install transformers')
4
  os.system('pip install scipy')
5
  os.system('pip install gradio')
 
 
 
 
 
 
 
 
6
  # Define the model path where the pre-trained model is saved on the Hugging Face model hub
7
  model_path = "Winnie-Kay/Finetuned_bert_model"
8
 
 
3
  os.system('pip install transformers')
4
  os.system('pip install scipy')
5
  os.system('pip install gradio')
6
+ import numpy as np
7
+ from scipy.special import softmax
8
+ import gradio as gr
9
+ from transformers import (
10
+ AutoTokenizer,
11
+ AutoConfig,
12
+ AutoModelForSequenceClassification,
13
+ TFAutoModelForSequenceClassification)
14
  # Define the model path where the pre-trained model is saved on the Hugging Face model hub
15
  model_path = "Winnie-Kay/Finetuned_bert_model"
16