LukeOLuck commited on
Commit
b02315a
·
1 Parent(s): 5d44125

update model inputs

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -34,10 +34,7 @@ def predict(text) -> Tuple[Dict, float]:
34
  model.eval()
35
 
36
  # Set up the inputs
37
- inputs = tokenizer(text, padding="max_length", truncation=True, return_tensors='pt')
38
-
39
- # Transform the input image for use with the model
40
- X = tokenizer(**inputs).unsqueeze(0) # unsqueeze = add batch dimension on 0th index
41
 
42
  # Put model into eval mode, make prediction
43
  model.eval()
 
34
  model.eval()
35
 
36
  # Set up the inputs
37
+ X = tokenizer(text, padding="max_length", truncation=True, return_tensors='pt')
 
 
 
38
 
39
  # Put model into eval mode, make prediction
40
  model.eval()