Update app.py

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