Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
3229c00
1
Parent(s):
97ab261
remove unused HfApi import from main.py
Browse files
main.py
CHANGED
@@ -15,7 +15,6 @@ from cashews import cache
|
|
15 |
from chromadb.utils import embedding_functions
|
16 |
from fastapi import FastAPI, HTTPException, Query
|
17 |
from fastapi.middleware.cors import CORSMiddleware
|
18 |
-
from huggingface_hub import HfApi, model_info
|
19 |
from pydantic import BaseModel
|
20 |
from transformers import AutoTokenizer
|
21 |
|
@@ -33,8 +32,6 @@ elif torch.backends.mps.is_available():
|
|
33 |
else:
|
34 |
DEVICE = "cpu"
|
35 |
|
36 |
-
hf_api = HfApi()
|
37 |
-
|
38 |
|
39 |
tokenizer = AutoTokenizer.from_pretrained(
|
40 |
"davanstrien/SmolLM2-360M-tldr-sft-2025-02-12_15-13"
|
|
|
15 |
from chromadb.utils import embedding_functions
|
16 |
from fastapi import FastAPI, HTTPException, Query
|
17 |
from fastapi.middleware.cors import CORSMiddleware
|
|
|
18 |
from pydantic import BaseModel
|
19 |
from transformers import AutoTokenizer
|
20 |
|
|
|
32 |
else:
|
33 |
DEVICE = "cpu"
|
34 |
|
|
|
|
|
35 |
|
36 |
tokenizer = AutoTokenizer.from_pretrained(
|
37 |
"davanstrien/SmolLM2-360M-tldr-sft-2025-02-12_15-13"
|