Berbex commited on
Commit
ce2b636
Β·
1 Parent(s): 877a87b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,10 +10,12 @@ from sklearn.metrics import f1_score, roc_auc_score, accuracy_score
10
  from transformers import EvalPrediction
11
  import torch
12
  import gradio as gr
 
13
  # REMOVE THIS IN COLAB
 
14
  title = 'Text market sentiment'
15
  text_ = "Bitcoin to the moon"
16
- model = AutoModelForSequenceClassification.from_pretrained("huggingface/Berbex/Model")
17
 
18
  inp = [gr.Textbox(label='API Key', placeholder="What is your API Key?"), gr.Textbox(label='Audio File URL', placeholder="Audio file URL?")]
19
  out = gr.Textbox(label='Output')
 
10
  from transformers import EvalPrediction
11
  import torch
12
  import gradio as gr
13
+
14
  # REMOVE THIS IN COLAB
15
+
16
  title = 'Text market sentiment'
17
  text_ = "Bitcoin to the moon"
18
+ model = torch.load("./model.pt")
19
 
20
  inp = [gr.Textbox(label='API Key', placeholder="What is your API Key?"), gr.Textbox(label='Audio File URL', placeholder="Audio file URL?")]
21
  out = gr.Textbox(label='Output')