Spaces:

codeparrot
/

code-complexity-predictor

Build error

loubnabnl HF Staff commited on Aug 25, 2022

Commit

b69dd7f

1 Parent(s): 950de6d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ model = AutoModelForSequenceClassification.from_pretrained("codeparrot/unixcoder
 def get_label(output):
     label = int(output[-1])
-    labels = ClassLabel(num_classes=7, names='constant', 'cubic', 'linear', 'logn', 'nlogn', 'np', 'quadratic'])
     return labels.int2str(label)
 def complexity_estimation(gen_prompt):

 def get_label(output):
     label = int(output[-1])
+    labels = ClassLabel(num_classes=7, names=['constant', 'cubic', 'linear', 'logn', 'nlogn', 'np', 'quadratic'])
     return labels.int2str(label)
 def complexity_estimation(gen_prompt):