skytnt commited on
Commit
073895e
1 Parent(s): 0d129bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=['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.Number(label="score"),title="Anime Aesthetic Predict",
27
  description='![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.anime-aesthetic-predict&countColor=%23263759&style=flat&labelStyle=lower)',
 
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='![Visitors](https://api.visitorbadge.io/api/visitors?path=skytnt.anime-aesthetic-predict&countColor=%23263759&style=flat&labelStyle=lower)',