Update src/core/inference.py

src/core/inference.py

@@ -0,0 +1,12 @@
+import streamlit as st
+from transformers import pipeline
+
+@st.cache_resource
+def load_deepfake_model():
+    """
+    Loads the Hugging Face pipeline for deepfake detection.
+    Using `@st.cache_resource` ensures the model is only loaded once per session.
+    """
+    model_name = "mangalaparida/dfimgdetection"
+    pipe = pipeline("image-classification", model=model_name)
+    return pipe