furquan commited on
Commit
6c4f963
1 Parent(s): 2ccd029

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def sentiment(text):
22
  tokenized = tokenizer(text, return_tensors='pt')
23
  with torch.no_grad():
24
  outputs = model.generate(
25
- input_ids=tokenized["input_ids"], attention_mask=tokenized["attention_mask"], max_new_tokens=1
26
  )
27
  return f"text: {text} Sentiment: {tokenizer.decode(outputs[0], skip_special_tokens=True).split(' ')[-1]}"
28
 
 
22
  tokenized = tokenizer(text, return_tensors='pt')
23
  with torch.no_grad():
24
  outputs = model.generate(
25
+ input_ids=tokenized["input_ids"], attention_mask=tokenized["attention_mask"]
26
  )
27
  return f"text: {text} Sentiment: {tokenizer.decode(outputs[0], skip_special_tokens=True).split(' ')[-1]}"
28