Update app.py

app.py

@@ -11,24 +11,6 @@ from transformers import EvalPrediction
 import torch
 import gradio as gr
 
-# REMOVE THIS IN COLAB
-
-title = 'Text market sentiment'
-text_ = "Bitcoin to the moon"
-model = torch.load("./model.pt", map_location=torch.device('cpu'))
-
-inp = [gr.Textbox(label='API Key', placeholder="What is your API Key?"), gr.Textbox(label='Audio File URL', placeholder="Audio file URL?")]
-out = gr.Textbox(label='Output')
-text_button = gr.Button("Flip")
-text_button.click(audio_to_text, inputs=inp, outputs=out)
-
-interface = gr.Interface.load(input=inp,output=out,
-title = title,
-theme = "peach",
-examples = [[text_]]).launch()
-
-###############
-
 console = Console()
 
 dataset = load_dataset("zeroshot/twitter-financial-news-sentiment", )
@@ -131,6 +113,52 @@ def compute_metrics(p: EvalPrediction):
     return result
 
 
+
+
+
+# REMOVE THIS IN COLAB #############
+
+title = 'Text market sentiment'
+text_ = "Bitcoin to the moon"
+model = torch.load("./model.pt", map_location=torch.device('cpu'))
+
+tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+
+text = "Bitcoin to the moon"
+
+encoding = tokenizer(text, return_tensors="pt")
+
+# apply sigmoid + threshold
+sigmoid = torch.nn.Sigmoid()
+probs = sigmoid(logits.squeeze().cpu())
+predictions = np.zeros(probs.shape)
+predictions[np.where(probs >= 0.5)] = 1
+# turn predicted id's into actual label names
+predicted_labels = [id2label[idx] for idx, label in enumerate(predictions) if label == 1.0]
+console.log("a")
+console.log(predicted_labels)
+console.log("a")
+
+
+
+
+
+inp = [gr.Textbox(label='API Key', placeholder="What is your API Key?"), gr.Textbox(label='Audio File URL', placeholder="Audio file URL?")]
+out = gr.Textbox(label='Output')
+text_button = gr.Button("Flip")
+text_button.click(audio_to_text, inputs=inp, outputs=out)
+
+interface = gr.Interface.load(input=inp,output=out,
+title = title,
+theme = "peach",
+examples = [[text_]]).launch()
+
+###############
+
+
+
+
+
 trainer = Trainer(
     model,
     args,