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='',
|