Spaces:
Sleeping
Sleeping
Shanshan Wang
commited on
Commit
·
953e634
1
Parent(s):
39b759d
try a preview h2o model
Browse files
app.py
CHANGED
|
@@ -10,7 +10,8 @@ import os
|
|
| 10 |
from huggingface_hub import login
|
| 11 |
hf_token = os.environ.get('hf_token', None)
|
| 12 |
|
| 13 |
-
path = 'h2oai/h2o-mississippi-2b'
|
|
|
|
| 14 |
|
| 15 |
# image preprocesing
|
| 16 |
IMAGENET_MEAN = (0.485, 0.456, 0.406)
|
|
|
|
| 10 |
from huggingface_hub import login
|
| 11 |
hf_token = os.environ.get('hf_token', None)
|
| 12 |
|
| 13 |
+
# path = 'h2oai/h2o-mississippi-2b'
|
| 14 |
+
path = "h2oai/h2o-mississippi1-2b-preview"
|
| 15 |
|
| 16 |
# image preprocesing
|
| 17 |
IMAGENET_MEAN = (0.485, 0.456, 0.406)
|