faceRecogModel / README.md
bytchew's picture
Updated package list
c193e24 verified
|
raw
history blame
1.31 kB
metadata
license: mit

explain a high-level overview of your code and how to run your code: including what packages I must install, and where the data came from and in which folder it must go.

Face Recognition Model

Files

  • modelWeights101.bin
    • Weights of model we trained using the Gebru server
  • app.py, model.py, and config.json
    • Standard files used by Hugging Face (HF) to setup the architecture of the model in conjunction with the weights
  • updateHFtemplate
    • Used this file to update the above files in the Hugging Face repository
    • The weights file did not update properly using standard gate, so we used Hugging Face Hub with this file to properly update the repo

Neccessary Package

  • Since the model all runs on Hugging Face, you don't need to install any packages.
  • However, if you do want to run the model locally, you will need to install these following packages:
    • torch
    • numpy
    • torchvision
    • huggingface_hub
    • You also need a personal HG login token to update using HF Hub

Running

  • You can run this model using the Hugging Face Space: CSSE416-final-project/faceRecogModel
    • You can upload images through the Space UI.
  • Alternatively, if you want to run the model locally, you can use <> file to load the weights.