ankush-003 commited on
Commit
b2d6b51
1 Parent(s): a8afce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -13,8 +13,10 @@ model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_
13
 
14
  def predict(username, pwd, label, payload_text = None):
15
  if(payload_text is None or payload_text is ""):
16
- payload["username"] = username
17
- payload["password"] = pwd
 
 
18
  payload_text = json.dumps(payload)
19
  inputs = tokenizer(payload_text, return_tensors="tf")
20
  # model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 
13
 
14
  def predict(username, pwd, label, payload_text = None):
15
  if(payload_text is None or payload_text is ""):
16
+ payload = {
17
+ "username": username,
18
+ "password": pwd
19
+ }
20
  payload_text = json.dumps(payload)
21
  inputs = tokenizer(payload_text, return_tensors="tf")
22
  # model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")