LuxExistentia commited on
Commit
443bcdd
1 Parent(s): 6001c23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ import gradio as gr
3
  import time
4
  from PIL import Image
5
  import os
 
6
 
7
- model_path = "deploying model/" + "vit_xsmall_patch16_clip_224(trainble_0.00) (F1 Score 96.83%, Loss 0.229548).onnx"
8
  input_size = [1, 3, 224, 224]
9
  img_size = input_size[-1]
10
 
 
3
  import time
4
  from PIL import Image
5
  import os
6
+ from pathlib import Path
7
 
8
+ model_path = list(Path("deploying model/").glob("*.onnx"))[0]
9
  input_size = [1, 3, 224, 224]
10
  img_size = input_size[-1]
11