davanstrien HF staff commited on
Commit
3229c00
·
1 Parent(s): 97ab261

remove unused HfApi import from main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -3
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"