Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import matplotlib.pyplot as plt
 import tensorflow as tf
+from huggingface_hub import from_pretrained_keras
 
-loaded_model = tf.saved_model.load("model/")
-loaded_model = loaded_model.signatures["serving_default"]
+loaded_model = from_pretrained_keras("alexanderkroner/MSI-Net")
 
 
 def get_target_shape(original_shape):