Create app.py

app.py

@@ -0,0 +1,44 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from huggingface_hub import HfApi, login
+import os
+
+app = FastAPI()
+
+# Replace with your Hugging Face token or set it as an environment variable
+HF_TOKEN = os.getenv("HF_TOKEN")
+REPO_ID = os.getenv("HF_REPO_ID")  # Replace with your model repo ID
+
+# Log in to Hugging Face once during app startup
+@app.on_event("startup")
+async def startup_event():
+    login(token=HF_TOKEN)
+
+@app.post("/upload/")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        # Initialize Hugging Face API
+        api = HfApi()
+
+        # Save the uploaded file temporarily
+        temp_file_path = f"/tmp/{file.filename}"
+        with open(temp_file_path, "wb") as temp_file:
+            temp_file.write(await file.read())
+
+        # Upload the file to the Hugging Face model repository
+        api.upload_file(
+            path_or_fileobj=temp_file_path,
+            path_in_repo=file.filename,  # File will be uploaded with its original name
+            repo_id=REPO_ID,
+            repo_type="model",
+        )
+
+        # Clean up temporary file
+        os.remove(temp_file_path)
+
+        return {"message": f"File '{file.filename}' uploaded successfully to {REPO_ID}"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Upload failed: {str(e)}")
+
+@app.get("/")
+async def root():
+    return {"message": "Welcome to the Hugging Face File Upload API"}