|
import gradio as gr |
|
import cv2 |
|
import base64 |
|
import requests |
|
|
|
def capture_video(): |
|
|
|
cap = cv2.VideoCapture(0) |
|
|
|
|
|
while True: |
|
|
|
ret, frame = cap.read() |
|
|
|
|
|
_, buffer = cv2.imencode('.jpg', frame) |
|
|
|
|
|
encoded_image = base64.b64encode(buffer).decode('utf-8') |
|
|
|
|
|
|
|
|
|
|
|
|
|
response = requests.post("https://abidlabs-pytorch-image-classifier.hf.space/api/predict", json={ |
|
"data": [(encoded_image) |
|
|
|
] |
|
}).json() |
|
print(response) |
|
|
|
|
|
cv2.imshow('frame', frame) |
|
|
|
|
|
if cv2.waitKey(1) & 0xFF == ord('q'): |
|
break |
|
|
|
|
|
cap.release() |
|
cv2.destroyAllWindows() |
|
|
|
iface = gr.Interface(fn=capture_video, inputs="webcam", outputs="text") |
|
iface.launch() |
|
|