Spaces:
Sleeping
Sleeping
okeowo1014
commited on
Commit
•
3b7916d
1
Parent(s):
4597f1c
Update main.py
Browse files
main.py
CHANGED
@@ -2,11 +2,10 @@ import io
|
|
2 |
from fastapi import FastAPI, File, UploadFile
|
3 |
import torch
|
4 |
import torchvision.transforms as transforms
|
5 |
-
import cv2
|
6 |
-
import numpy as np
|
7 |
from PIL import Image
|
8 |
import torch.nn.functional as F
|
9 |
import torch.nn as nn
|
|
|
10 |
num_classes = 10
|
11 |
|
12 |
# Class definition for the model (same as in your code)
|
@@ -69,3 +68,4 @@ async def predict_endpoint(file: UploadFile = File(...)):
|
|
69 |
class_labels = {0: 'Left_ring_fingers', 1: 'Left_thumb_fingers', 2: 'Right_index_fingers', 3: 'Right_little_fingers', 4: 'Right_middle_fingers', 5: 'Right_ring_fingers', 6: 'Right_thumb_fingers', 7: 'left_index_fingers', 8: 'left_little_fingers', 9: 'left_middle_fingers'}
|
70 |
return {"predicted_class": predicted_class, "class_label": class_labels[predicted_class]}
|
71 |
|
|
|
|
2 |
from fastapi import FastAPI, File, UploadFile
|
3 |
import torch
|
4 |
import torchvision.transforms as transforms
|
|
|
|
|
5 |
from PIL import Image
|
6 |
import torch.nn.functional as F
|
7 |
import torch.nn as nn
|
8 |
+
|
9 |
num_classes = 10
|
10 |
|
11 |
# Class definition for the model (same as in your code)
|
|
|
68 |
class_labels = {0: 'Left_ring_fingers', 1: 'Left_thumb_fingers', 2: 'Right_index_fingers', 3: 'Right_little_fingers', 4: 'Right_middle_fingers', 5: 'Right_ring_fingers', 6: 'Right_thumb_fingers', 7: 'left_index_fingers', 8: 'left_little_fingers', 9: 'left_middle_fingers'}
|
69 |
return {"predicted_class": predicted_class, "class_label": class_labels[predicted_class]}
|
70 |
|
71 |
+
|