ariG23498 commited on
Commit
c01f167
1 Parent(s): ce34875

chore: take the model form keras-io

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import tensorflow as tf
3
  import gradio as gr
4
 
5
  # download the model in the global context
6
- vis_model = from_pretrained_keras("ariG23498/involution")
7
 
8
  def infer(test_image):
9
  # convert the image to a tensorflow tensor and resize the image
 
3
  import gradio as gr
4
 
5
  # download the model in the global context
6
+ vis_model = from_pretrained_keras("keras-io/involution")
7
 
8
  def infer(test_image):
9
  # convert the image to a tensorflow tensor and resize the image