Spaces:
Running
Running
add examples
Browse files- app.py +3 -1
- examples/00.jpg +0 -0
- examples/01.jpg +0 -0
app.py
CHANGED
@@ -22,5 +22,7 @@ def predict(img):
|
|
22 |
if __name__ == "__main__":
|
23 |
model_path = hf_hub_download(repo_id="skytnt/anime_quality", filename="classifier.onnx")
|
24 |
model = rt.InferenceSession(model_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
|
25 |
-
|
|
|
|
|
26 |
app.launch()
|
|
|
22 |
if __name__ == "__main__":
|
23 |
model_path = hf_hub_download(repo_id="skytnt/anime_quality", filename="classifier.onnx")
|
24 |
model = rt.InferenceSession(model_path, providers=['CUDAExecutionProvider', '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.Label(label="result"),title="Best Anime or Not",
|
27 |
+
allow_flagging="never", examples=examples)
|
28 |
app.launch()
|
examples/00.jpg
ADDED
examples/01.jpg
ADDED