Spaces:
Sleeping
Sleeping
File size: 1,365 Bytes
03fc4f2 f68d424 03fc4f2 f68d424 03fc4f2 f68d424 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
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
def get_color_labels(guesses: List[str], actual_label: Optional[str]) -> List[str]:
if not actual_label:
return ["white"] * len(guesses)
return ["lime" if x == actual_label else "white" for x in guesses]
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):
top_guesses = predict_url(input_img)
elif isinstance(input_img, (InMemoryUploadedFile, TemporaryUploadedFile)):
input_img.seek(0)
top_guesses = predict_file(input_img.read())
else:
logging.error(f"Unknown input type: {type(input_img)=}")
top_guesses = ["Unknown Input Type", "", "", "", ""]
color_labels = get_color_labels(top_guesses, actual_label)
return top_guesses, color_labels
def is_healthy() -> bool:
return True
|