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