Refactor name to huggingface_client

src/model_client.py

@@ -1,15 +1,10 @@
-from time import time
 from typing import List, Optional, Tuple, Union
-import json
-import os
-import requests
-from urllib.parse import urljoin
 
 from django.core.files.uploadedfile import InMemoryUploadedFile, TemporaryUploadedFile
 
 import logging
 
-from huggingface.predict import predict_file, predict_url
+from huggingface_client.predict import predict_file, predict_url
 
 
 
@@ -23,10 +18,7 @@ def url_image_vars(
     input_img: Union[str, InMemoryUploadedFile, TemporaryUploadedFile], label: str
 ) -> Tuple[List[str], List[str]]:
     actual_label = label.title()
-    if not is_healthy():
-        logging.error("Model failed healthcheck")
-        top_guesses = ["Model Offline", "Try Again Later", "", "", ""]
-    elif isinstance(input_img, str):
+    if isinstance(input_img, str):
         top_guesses = predict_url(input_img)
     elif isinstance(input_img, (InMemoryUploadedFile, TemporaryUploadedFile)):
         input_img.seek(0)
@@ -37,6 +29,3 @@ def url_image_vars(
     color_labels = get_color_labels(top_guesses, actual_label)
     return top_guesses, color_labels
 
-
-def is_healthy() -> bool:
-    return True