Update app.py

app.py

@@ -4,6 +4,8 @@ import numpy
 import tensorflow
 import math
 from tensorflow.python.framework.ops import disable_eager_execution
+import huggingface_hub # for loading model
+
 
 # Because important
 disable_eager_execution()
@@ -233,18 +235,20 @@ def generate_unit_cell(t, d, th):
 
 def interpolate(t1, t2, d1, d2, th1, th2, steps):
     # Load the decoder model
-    decoder_model_boxes = tensorflow.keras.models.load_model('data/decoder_model_boxes', compile=False)
+    decoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-decoder")
+    # decoder_model_boxes = tensorflow.keras.models.load_model('data/decoder_model_boxes', compile=False)
 
     # # import the encoder model architecture
-    json_file_loaded = open('data/model.json', 'r')
-    loaded_model_json = json_file_loaded.read()
+    # json_file_loaded = open('data/model.json', 'r')
+    # loaded_model_json = json_file_loaded.read()
 
     # load model using the saved json file
-    encoder_model_boxes = tensorflow.keras.models.model_from_json(loaded_model_json)
+    # encoder_model_boxes = tensorflow.keras.models.model_from_json(loaded_model_json)
 
     # load weights into newly loaded_model
-    encoder_model_boxes.load_weights('data/model_tf')
-
+    # encoder_model_boxes.load_weights('data/model_tf')
+    encoder_model_boxes = huggingface_hub.from_pretrained_keras("cmudrc/2d-lattice-encoder")
+    
     num_internal = int(steps)
     number_1 = generate_unit_cell(t1, d1, th1)
     number_2 = generate_unit_cell(t2, d2, th2)