Hugging Face's logo

Spaces:
ethanrom
/
ethanrom-a2
Runtime error

App Files Community
ethanrom commited on
Commit
cdf5643
•
1 Parent(s): bca5a82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def predict_sentiment(text_input, model_selection):
29
  end_time = time.time()
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 = [
 
29
  end_time = time.time()
30
  predicted_class = result["labels"][0]
31
  inference_time = end_time - start_time
32
+ model_size = pretrained_model.model.num_parameters()
33
  return predicted_class, inference_time, model_size
34
 
35
  inputs = [