Spaces:

skytnt
/

anime-aesthetic-predict

Running

skytnt commited on Jun 13, 2024

Commit

66a7855

1 Parent(s): e48d82b

visitorbadge

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,5 +24,6 @@ if __name__ == "__main__":
     model = rt.InferenceSession(model_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
     examples = [[f"examples/{x:02d}.jpg"] for x in range(0, 2)]
     app = gr.Interface(predict, gr.Image(label="input image"), gr.Number(label="score"),title="Anime Aesthetic Predict",
                        allow_flagging="never", examples=examples, cache_examples=False)
     app.launch()

     model = rt.InferenceSession(model_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
     examples = [[f"examples/{x:02d}.jpg"] for x in range(0, 2)]
     app = gr.Interface(predict, gr.Image(label="input image"), gr.Number(label="score"),title="Anime Aesthetic Predict",
+                       description='![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.anime-aesthetic-predict&countColor=%23263759&style=flat&labelStyle=lower)',
                        allow_flagging="never", examples=examples, cache_examples=False)
     app.launch()