Hugging Face's logo

Spaces:
keras-io
/
video-vision-transformer-CT
Build error

App Files Community
1
pablorodriper commited on
Commit
a32722f
•
1 Parent(s): 66605df

Change model

Browse files
Files changed (1) hide show
  1. utils/predict.py +1 -1
utils/predict.py CHANGED
@@ -10,7 +10,7 @@ def get_model():
10
  """
11
  Download the model from the Hugging Face Hub and compile it.
12
  """
13
- model = from_pretrained_keras("pablorodriper/video-vision-transformer")
14
 
15
  model.compile(
16
  optimizer=Adam(learning_rate=LEARNING_RATE),
 
10
  """
11
  Download the model from the Hugging Face Hub and compile it.
12
  """
13
+ model = from_pretrained_keras("keras-io/video-vision-transformer")
14
 
15
  model.compile(
16
  optimizer=Adam(learning_rate=LEARNING_RATE),