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 |
|