Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def predict_sentiment(text_input, model_selection):
|
|
30 |
predicted_class = result["labels"][0]
|
31 |
inference_time = end_time - start_time
|
32 |
model_size = pretrained_tokenizer.model_max_length + pretrained_model.model.num_parameters()
|
33 |
-
return
|
34 |
|
35 |
inputs = [
|
36 |
gr.inputs.Textbox("Enter text"),
|
|
|
30 |
predicted_class = result["labels"][0]
|
31 |
inference_time = end_time - start_time
|
32 |
model_size = pretrained_tokenizer.model_max_length + pretrained_model.model.num_parameters()
|
33 |
+
return predicted_class, inference_time, model_size
|
34 |
|
35 |
inputs = [
|
36 |
gr.inputs.Textbox("Enter text"),
|