huimanho commited on
Commit
4b4ac3c
1 Parent(s): bfbb65f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def main():
15
  num_labels=5)
16
  sentiment_pipeline = pipeline(model="huimanho/CustomModel_yelp")
17
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
18
- outputs = model2(**inputs)
19
  predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
20
  predictions = predictions.cpu().detach().numpy()
21
  # Get the index of the largest output value
 
15
  num_labels=5)
16
  sentiment_pipeline = pipeline(model="huimanho/CustomModel_yelp")
17
  tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
18
+ outputs = model2(**user_input)
19
  predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
20
  predictions = predictions.cpu().detach().numpy()
21
  # Get the index of the largest output value