|
import streamlit as st |
|
from PIL import Image |
|
import requests |
|
from transformers import pipeline |
|
|
|
|
|
model_name = "AP4556/Do-Model" |
|
pipe = pipeline('image-classification', model=model_name) |
|
|
|
st.title("Deepfake vs Real Image Detection") |
|
|
|
uploaded_file = st.file_uploader("Choose an image...", type="jpg") |
|
if uploaded_file is not None: |
|
image = Image.open(uploaded_file) |
|
st.image(image, caption='Uploaded Image.', use_column_width=True) |
|
st.write("") |
|
st.write("Classifying...") |
|
|
|
|
|
result = pipe(image) |
|
|
|
|
|
st.write(result) |