Update backend.py
Browse files- backend.py +9 -4
backend.py
CHANGED
|
@@ -5,13 +5,18 @@ from PIL import Image, UnidentifiedImageError
|
|
| 5 |
from io import BytesIO
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
import urllib3
|
| 8 |
-
from transformers import pipeline
|
|
|
|
| 9 |
import torch
|
| 10 |
import textwrap
|
| 11 |
-
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
from huggingface_hub import InferenceClient
|
| 14 |
-
|
| 15 |
IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
|
| 16 |
IS_SPACE = os.environ.get("SPACE_ID", None) is not None
|
| 17 |
|
|
|
|
| 5 |
from io import BytesIO
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
import urllib3
|
| 8 |
+
from transformers import pipeline
|
| 9 |
+
from transformers import BitsAndBytesConfig
|
| 10 |
import torch
|
| 11 |
import textwrap
|
| 12 |
+
import pandas as pd
|
| 13 |
+
import numpy as np
|
| 14 |
+
from haversine import haversine # Install haversine library: pip install haversine
|
| 15 |
+
from transformers import AutoProcessor, LlavaForConditionalGeneration
|
| 16 |
+
from transformers import BitsAndBytesConfig
|
| 17 |
+
import torch
|
| 18 |
from huggingface_hub import InferenceClient
|
| 19 |
+
|
| 20 |
IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
|
| 21 |
IS_SPACE = os.environ.get("SPACE_ID", None) is not None
|
| 22 |
|