Lambang commited on
Commit
a3de917
·
1 Parent(s): 2c2434b

update req

Browse files
Files changed (2) hide show
  1. main.py +9 -12
  2. requirements.txt +0 -0
main.py CHANGED
@@ -24,6 +24,7 @@ from train_pred import TrainPred
24
  app = FastAPI()
25
  public_url = "https://lambang0902-test-space.hf.space"
26
  app.mount("/static", StaticFiles(directory="static"), name="static")
 
27
  # Tempat deklarasi variabel-variabel penting
28
  filepath = ""
29
  list_class = ['Diamond','Oblong','Oval','Round','Square','Triangle']
@@ -31,28 +32,24 @@ list_folder = ['Training', 'Testing']
31
  face_crop_img = True
32
  face_landmark_img = True
33
  landmark_extraction_img = True
34
- # #-----------------------------------------------------
35
- #
36
- #
37
- # #-----------------------------------------------------
38
  # Tempat deklarasi model dan sejenisnya
39
  selected_model = tf.keras.models.load_model(f'models/fc_model_1.h5', compile=False)
40
  face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_alt2.xml')
41
  mp_drawing = mp.solutions.drawing_utils
42
  mp_face_mesh = mp.solutions.face_mesh
43
  drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
44
- # #-----------------------------------------------------
45
- #
46
- #
47
- # #-----------------------------------------------------
48
  # Tempat setting server
49
  UPLOAD_FOLDER = './upload'
50
  UPLOAD_MODEL = './models'
51
  ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg','zip','h5'}
52
- # app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
53
- # app.config['UPLOAD_MODEL'] = UPLOAD_MODEL
54
- # app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024 # 500 MB
55
- # #-----------------------------------------------------
56
  #Endpoints
57
  #Root endpoints
58
  @app.get("/")
 
24
  app = FastAPI()
25
  public_url = "https://lambang0902-test-space.hf.space"
26
  app.mount("/static", StaticFiles(directory="static"), name="static")
27
+
28
  # Tempat deklarasi variabel-variabel penting
29
  filepath = ""
30
  list_class = ['Diamond','Oblong','Oval','Round','Square','Triangle']
 
32
  face_crop_img = True
33
  face_landmark_img = True
34
  landmark_extraction_img = True
35
+ # -----------------------------------------------------
36
+
37
+ # -----------------------------------------------------
 
38
  # Tempat deklarasi model dan sejenisnya
39
  selected_model = tf.keras.models.load_model(f'models/fc_model_1.h5', compile=False)
40
  face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_alt2.xml')
41
  mp_drawing = mp.solutions.drawing_utils
42
  mp_face_mesh = mp.solutions.face_mesh
43
  drawing_spec = mp_drawing.DrawingSpec(thickness=1, circle_radius=1)
44
+ # -----------------------------------------------------
45
+
46
+
47
+ # -----------------------------------------------------
48
  # Tempat setting server
49
  UPLOAD_FOLDER = './upload'
50
  UPLOAD_MODEL = './models'
51
  ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg','zip','h5'}
52
+ # -----------------------------------------------------
 
 
 
53
  #Endpoints
54
  #Root endpoints
55
  @app.get("/")
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ