Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ def predict(img):
|
|
| 21 |
|
| 22 |
if __name__ == "__main__":
|
| 23 |
model_path = hf_hub_download(repo_id="skytnt/anime-aesthetic", filename="model.onnx")
|
| 24 |
-
model = rt.InferenceSession(model_path, providers=['
|
| 25 |
examples = [[f"examples/{x:02d}.jpg"] for x in range(0, 2)]
|
| 26 |
app = gr.Interface(predict, gr.Image(label="input image"), gr.Number(label="score"),title="Anime Aesthetic Predict",
|
| 27 |
description='',
|
|
|
|
| 21 |
|
| 22 |
if __name__ == "__main__":
|
| 23 |
model_path = hf_hub_download(repo_id="skytnt/anime-aesthetic", filename="model.onnx")
|
| 24 |
+
model = rt.InferenceSession(model_path, providers=['CPUExecutionProvider'])
|
| 25 |
examples = [[f"examples/{x:02d}.jpg"] for x in range(0, 2)]
|
| 26 |
app = gr.Interface(predict, gr.Image(label="input image"), gr.Number(label="score"),title="Anime Aesthetic Predict",
|
| 27 |
description='',
|