Hugging Face's logo

Spaces:
Nirav-Madhani
/
Octo-1.5-Base
Sleeping

App Files Community
Nirav-Madhani commited on
Commit
8a209a5
·
verified ·
1 Parent(s): 89208ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from fastapi.openapi.docs import get_swagger_ui_html
14
  os.environ['JAX_PLATFORMS'] = 'cpu'
15
 
16
  # Load the model once globally
17
- model = OctoModel.load_pretrained("hf://rail-berkeley/octo-small-1.5")
18
 
19
  # Initialize FastAPI app
20
  app = FastAPI(
 
14
  os.environ['JAX_PLATFORMS'] = 'cpu'
15
 
16
  # Load the model once globally
17
+ model = OctoModel.load_pretrained("hf://rail-berkeley/octo-base-1.5")
18
 
19
  # Initialize FastAPI app
20
  app = FastAPI(