Spaces:

ankush-003
/

ankush-003-nosqli_identifier

Runtime error

ankush-003 commited on Jul 12, 2023

Commit

a8afce4

•

1 Parent(s): b280351

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 def predict(username, pwd, label, payload_text = None):
-    if(payload_text is None or payload_text = ""):
         payload["username"] = username
         payload["password"] = pwd
         payload_text = json.dumps(payload)

 model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 def predict(username, pwd, label, payload_text = None):
+    if(payload_text is None or payload_text is ""):
         payload["username"] = username
         payload["password"] = pwd
         payload_text = json.dumps(payload)