disaster type class

app.py

@@ -40,8 +40,10 @@ def disaster_classification(img):
     image[0] = img
     # prediction = model.predict(image).tolist()[0]
     # prediction = np.random.rand(1, 6)[0]
+    print('hello')
     results = json.loads(rest_request(image, get_rest_url(model_name='disaster-classification-model', host='3.86.228.238')).content)
     prediction = results['predictions'][0]
+    print('hello back')
 
     return {disaster_types[i]: prediction[i] for i in range(len(disaster_types))}