kalpniks commited on
Commit
0fbdd1f
·
verified ·
1 Parent(s): 03616e9

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -3,8 +3,7 @@ import os
3
  from collections import Counter
4
  import time
5
  import traceback
6
- from transformers import AutoImageProcessor
7
- from transformers import SiglipForImageClassification
8
  from PIL import Image
9
  import torch
10
  import cv2
@@ -70,7 +69,7 @@ class SignLanguageVideoProcessor(VideoProcessorBase):
70
 
71
  inputs = self.processor(images=img_pil, return_tensors="pt")
72
  with torch.no_grad():
73
- outputs = self.model(**inputs)
74
  logits = outputs.logits
75
 
76
  predicted_label_index = torch.argmax(logits, dim=1).item()
 
3
  from collections import Counter
4
  import time
5
  import traceback
6
+ from transformers import AutoImageProcessor, SiglipForImageClassification
 
7
  from PIL import Image
8
  import torch
9
  import cv2
 
69
 
70
  inputs = self.processor(images=img_pil, return_tensors="pt")
71
  with torch.no_grad():
72
+ outputs = model(**inputs)
73
  logits = outputs.logits
74
 
75
  predicted_label_index = torch.argmax(logits, dim=1).item()
requirements.txt CHANGED
@@ -7,4 +7,4 @@ Pillow
7
  streamlit_webrtc
8
  av==12.0.0
9
  numpy
10
- huggingface_hub==0.20.0
 
7
  streamlit_webrtc
8
  av==12.0.0
9
  numpy
10
+ # huggingface_hub==0.20.0 # Removed pinning to let transformers choose