Update app.py

app.py

@@ -6,9 +6,10 @@ from starlette.responses import StreamingResponse
 from fastapi.responses import FileResponse
 import io
 from fastapi.middleware.cors import CORSMiddleware
-from config import settings
+import os
+from dotenv import load_dotenv
 
-login(settings.huggingface_key)
+HF_TOKEN = os.getenv('HF_TOKEN')
 app = FastAPI()
 
 app.add_middleware(
@@ -26,7 +27,7 @@ else:
     device = "cuda" if torch.cuda.is_available() else "cpu"
     
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
-pipe1 = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5").to(device)
+pipe1 = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5,", use_auth_token=HF_TOKEN).to(device)
 
 @app.get("/")
 async def root():