albumup commited on
Commit
fef6019
·
verified ·
1 Parent(s): d889136

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -46,10 +46,10 @@ async def handle_upload(file: UploadFile = File(...)):
46
  if not upload_success:
47
  return JSONResponse({"error": "Upload failed"}, status_code=500)
48
 
49
- mirrored_url = f"/rbxg/{upload_result['serving_url'].split('/pbxt/')[1]}"
50
- return JSONResponse({"url": mirrored_url})
51
 
52
- @app.get("/rbxg/{path:path}")
53
  async def handle_stream(path: str, request: Request):
54
  original_url = f'https://replicate.delivery/pbxt/{path}'
55
  range_header = request.headers.get('Range')
@@ -97,3 +97,7 @@ async def retry_upload(upload_url, file_content, content_type, max_retries=5):
97
  await asyncio.sleep(delay)
98
  delay *= 2
99
  return False
 
 
 
 
 
46
  if not upload_success:
47
  return JSONResponse({"error": "Upload failed"}, status_code=500)
48
 
49
+ direct_url = f"https://replicate.delivery/pbxt/{upload_result['serving_url'].split('/pbxt/')[1]}"
50
+ return JSONResponse({"url": direct_url})
51
 
52
+ @app.get("/file/{path:path}")
53
  async def handle_stream(path: str, request: Request):
54
  original_url = f'https://replicate.delivery/pbxt/{path}'
55
  range_header = request.headers.get('Range')
 
97
  await asyncio.sleep(delay)
98
  delay *= 2
99
  return False
100
+
101
+ if __name__ == "__main__":
102
+ import uvicorn
103
+ uvicorn.run(app, host="0.0.0.0", port=8000)