Update main.py

main.py

@@ -16,10 +16,10 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
+HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Initialize Gradio client
-hf_token = os.environ.get('HF_TOKEN')
-client = Client("https://Makhinur/Image_Face_Upscale_Restoration-GFPGAN.hf.space/", hf_token=hf_token)
+# Initialize the Gradio client with the token
+client = Client("Makhinur/Image_Face_Upscale_Restoration-GFPGAN", hf_token=HF_TOKEN)
 
 @app.post("/upload/")
 async def upload_file(file: UploadFile = File(...), version: str = Form(...), scale: int = Form(...)):