Hugging Face's logo

Spaces:
Neurolingua
/
AgriChatbot
Sleeping

App Files Community
Neurolingua commited on
Commit
e9ec5c3
•
1 Parent(s): e9086c4

Update other_function.py

Browse files
Files changed (1) hide show
  1. other_function.py +1 -1
other_function.py CHANGED
@@ -38,7 +38,7 @@ def predict_pest(filepath):
38
  api_key="oF1aC4b1FBCDtK8CoKx7"
39
  )
40
  result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
41
- return result['predicted_classes'][0]
42
 
43
 
44
  def predict_disease(filepath):
 
38
  api_key="oF1aC4b1FBCDtK8CoKx7"
39
  )
40
  result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
41
+ return result['predictions'][0]
42
 
43
 
44
  def predict_disease(filepath):