app.py

import gradio as gr
from transformers import pipeline

# Load the models using pipeline
image_model = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")

# Define the prediction function
def predict(image):
 result = image_model(image)
 print("Raw prediction result:", result) # Debugging statement
 # Convert the result to the expected format
 output = {item['label']: item['score'] for item in result}
 print("Formatted prediction result:", output) # Debugging statement
 return output

# Create Gradio interface
inputs = gr.Image(type="filepath", label="Upload Image File", visible=False)

outputs = gr.Label(output)

gr.Interface(fn=predict, inputs=image_input, outputs=output).launch()