Spaces:

CesarLeblanc
/

plantbert_space

Runtime error

CesarLeblanc commited on Nov 14, 2023

Commit

54a8fdb

•

1 Parent(s): 443a3b3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ def plantbert(text, typology, confidence, task):
     if task == "classification":
         formatted_output, image_output = classification(text, typology, confidence, task)
     else:
-        formatted_output, image_output = masking(text, typology, confidence, task)
     return formatted_output, image_output
 inputs=[

     if task == "classification":
         formatted_output, image_output = classification(text, typology, confidence, task)
     else:
+        formatted_output, image_output = masking(text, task)
     return formatted_output, image_output
 inputs=[