Update other_function.py

other_function.py

@@ -61,21 +61,27 @@ class ConversationBufferMemory:
     def get_memory(self):
         return self.memory
 def predict_pest(filepath):
-    CLIENT = InferenceHTTPClient(
-        api_url="https://detect.roboflow.com",
-        api_key="oF1aC4b1FBCDtK8CoKx7"
-    )
-    result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
-    return result['predictions'][0]
+    try:
+        CLIENT = InferenceHTTPClient(
+            api_url="https://detect.roboflow.com",
+            api_key="oF1aC4b1FBCDtK8CoKx7"
+        )
+        result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
+        return result['predictions'][0]
+    except:
+        return None
     
 
 def predict_disease(filepath):
-    CLIENT = InferenceHTTPClient(
+    try:
+        CLIENT = InferenceHTTPClient(
         api_url="https://classify.roboflow.com",
         api_key="oF1aC4b1FBCDtK8CoKx7"
     )
-    result = CLIENT.infer(filepath, model_id="plant-disease-detection-iefbi/1")
-    return result['predicted_classes'][0]
+        result = CLIENT.infer(filepath, model_id="plant-disease-detection-iefbi/1")
+        return result['predicted_classes'][0]
+    except:
+        return None
 
 
 def convert_img(url, account_sid, auth_token):