Update app.py

app.py

@@ -2,6 +2,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import numpy as np
+import gradio as gr
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
@@ -11,6 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
 model = model.to(device)
 
+query = 'I am not having a great day.'
 inputs = tokenizer(query, return_tensors='pt', truncation=True)
 inputs = inputs.to(device)
 outputs = model(**inputs)
@@ -45,7 +47,7 @@ def get_predictions(input_text: str) -> dict:
     label2id = {k: float(v) for k, v in sorted(label2id.items(), key=lambda item: item[1].item(), reverse=True)}
     return label2id
     
-import gradio as gr
+
 gr.Interface(
     fn=get_predictions,
     inputs=gr.components.Textbox(label='Input'),