Spaces:
Sleeping
Sleeping
Commit
·
d1625ff
1
Parent(s):
0114d4f
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,11 +7,11 @@ api = hf_api.HfApi()
|
|
| 7 |
flyswot_models = ModelFilter(author="flyswot")
|
| 8 |
models = api.list_models(filter=flyswot_models)
|
| 9 |
models = [model.modelId for model in models]
|
| 10 |
-
choices = gr.inputs.Dropdown(choices=models, default='flyswot/convnext-tiny-224_flyswot',label="Model")
|
| 11 |
|
|
|
|
| 12 |
|
| 13 |
-
def predict(image
|
| 14 |
-
pipe = pipeline("image-classification", model=model)
|
| 15 |
predictions = pipe(image, top_k=20)
|
| 16 |
return {pred['label']: pred['score'] for pred in predictions}
|
| 17 |
|
|
|
|
| 7 |
flyswot_models = ModelFilter(author="flyswot")
|
| 8 |
models = api.list_models(filter=flyswot_models)
|
| 9 |
models = [model.modelId for model in models]
|
| 10 |
+
#choices = gr.inputs.Dropdown(choices=models, default='flyswot/convnext-tiny-224_flyswot',label="Model")
|
| 11 |
|
| 12 |
+
pipe = pipeline("image-classification", model="flyswot/convnext-tiny-224_flyswot")
|
| 13 |
|
| 14 |
+
def predict(image):
|
|
|
|
| 15 |
predictions = pipe(image, top_k=20)
|
| 16 |
return {pred['label']: pred['score'] for pred in predictions}
|
| 17 |
|