Alesteba commited on
Commit
e89796e
·
1 Parent(s): de5aa49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -241,11 +241,15 @@ st.markdown("## Interactive Demo")
241
 
242
  # download the model:
243
 
244
- from huggingface_hub import snapshot_download
245
- snapshot_download(repo_id="Alesteba/your-model-name", local_dir="./nerf")
 
 
 
 
246
 
247
  # load the pre-trained model
248
- nerf_loaded = tf.keras.models.load_model("nerf", compile=False)
249
 
250
  # set the values of r theta phi
251
  r = 4.0
 
241
 
242
  # download the model:
243
 
244
+ # from huggingface_hub import snapshot_download
245
+ # snapshot_download(repo_id="Alesteba/your-model-name", local_dir="./nerf")
246
+
247
+ from huggingface_hub import from_pretrained_keras
248
+
249
+ nerf_loaded = from_pretrained_keras("Alesteba/NeRF_ficus")
250
 
251
  # load the pre-trained model
252
+ # nerf_loaded = tf.keras.models.load_model("nerf", compile=False)
253
 
254
  # set the values of r theta phi
255
  r = 4.0