Merge branch 'main' of https://huggingface.co/spaces/artglobal/minicpm-fastapi

app.py

@@ -1,3 +1,5 @@
+import os
+
 from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 from typing import List
 from pydantic import BaseModel
@@ -5,6 +7,13 @@ from PIL import Image
 import io
 from transformers import AutoModel, AutoTokenizer
 import torch
+from huggingface_hub import login
+from dotenv import load_dotenv
+load_dotenv()
+
+api_key = os.getenv("HF_TOKEN")
+
+login(token = api_key)
 
 app = FastAPI()
 

requirements.txt

@@ -6,4 +6,6 @@ torchvision==0.16.2
 transformers==4.40.0
 sentencepiece==0.1.99
 decord
-python-multipart
+python-multipart
+huggingface_hub
+python-dotenv