Error handling

app.py

@@ -129,8 +129,8 @@ def predict(tweet):
 
     if type(prediction) == dict: 
         print(prediction)
-        loading_time = prediction['estimated_time']
-        return f"Loading Model (Estimated Time: {loading_time} Seconds)"
+        error_message = prediction['error']
+        return error_message
 
     
     if bool(matched_profanity) == False: