turhancan97 commited on
Commit
f1a7938
1 Parent(s): eb45374

Update MAE-ViT image reconstruction descriptions and add links to model card and GitHub repository

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -209,7 +209,7 @@ inference_no_mode = gr.Interface(
209
  examples=path_1,
210
  cache_examples = False,
211
  title="MAE-ViT Image Reconstruction",
212
- description="This is a demo of the MAE-ViT model for image reconstruction. The model is trained without PCA mode. It was trained on the STL-10 dataset.",
213
  )
214
 
215
  inference_pca_mode_bottom = gr.Interface(
 
209
  examples=path_1,
210
  cache_examples = False,
211
  title="MAE-ViT Image Reconstruction",
212
+ description="This is a demo of the MAE-ViT model for image reconstruction. The model is trained without PCA mode. It was trained on the STL-10 dataset. Check out the huggingface model card and the github repository for more information. https://huggingface.co/turhancan97/MAE-Models and https://github.com/turhancan97/Learning-by-Reconstruction-with-MAE",
213
  )
214
 
215
  inference_pca_mode_bottom = gr.Interface(