Upload 2 files

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from gradio_client import Client , handle_file
+import cv2
+import os
+from PIL import Image
+
+clientImgPipeLn = Client("dj-dawgs-ipd/IPD_IMAGE_PIPELINE")
+
+def predict(video_path):
+    cap = cv2.VideoCapture(video_path)
+    fps = int(cap.get(cv2.CAP_PROP_FPS))  
+    frame_interval = fps * 2
+
+    frame_count = 0
+    success = True
+
+    temp_dir = "temp_frames"
+    os.makedirs(temp_dir, exist_ok=True)
+    
+    res = 'not_hate'
+
+    while success:
+        success, frame = cap.read()
+        if frame_count % frame_interval == 0 and success:
+            temp_image_path = os.path.join(temp_dir, f"frame_{frame_count // fps}s.jpg")
+            cv2.imwrite(temp_image_path, frame)
+
+            response = clientImgPipeLn.predict(
+                image=handle_file(temp_image_path),
+                api_name="/predict"
+            )
+            print(f"Response for frame at {frame_count // fps}s: {response}")
+            if(response[0]['label'] == 'hate'):
+                res = 'hate'
+                break
+
+        frame_count += 1
+
+    cap.release()
+
+    for file in os.listdir(temp_dir):
+        os.remove(os.path.join(temp_dir, file))
+    os.rmdir(temp_dir)
+
+    print("prediction successful")
+    
+    return res
+    
+iface = gr.Interface(fn=predict,
+                     inputs = gr.Video(),
+                     outputs=[gr.Label(label = "Class")],
+                     title = "Hate Speech Detection in Video",
+                     description = "Detect hateful symbols or text in Video"
+)
+                     
+if __name__ == "__main__":
+    iface.launch()
+

requirements.txt

@@ -0,0 +1,53 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.7.0
+certifi==2024.12.14
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+fastapi==0.115.6
+ffmpy==0.5.0
+filelock==3.16.1
+fsspec==2024.12.0
+gradio==5.9.1
+gradio_client==1.5.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.0
+idna==3.10
+Jinja2==3.1.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+numpy==2.2.1
+opencv-python==4.10.0.84
+orjson==3.10.12
+packaging==24.2
+pandas==2.2.3
+pillow==11.0.0
+pydantic==2.10.4
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.18.0
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2024.2
+PyYAML==6.0.2
+requests==2.32.3
+rich==13.9.4
+ruff==0.8.4
+safehttpx==0.1.6
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.41.3
+tomlkit==0.13.2
+tqdm==4.67.1
+typer==0.15.1
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==14.1