Spaces:

Neurolingua
/

AgriChatbot

App Files Files Community

Neurolingua commited on Aug 26

Commit

a4684f0

•

1 Parent(s): 8cc48a4

Update other_function.py

Files changed (1) hide show

other_function.py +4 -1

other_function.py CHANGED Viewed

@@ -67,7 +67,10 @@ def predict_pest(filepath):
             api_key="oF1aC4b1FBCDtK8CoKx7"
         )
         result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
-        return result['predictions'][0]
     except:
         return None

             api_key="oF1aC4b1FBCDtK8CoKx7"
         )
         result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
+        a= result['predictions'][0]
+        if a=='x':
+            return 'APHIDS'
+        return a
     except:
         return None