Spaces:

allimonae
/

basic-AI-image-detector

Running

allimonae commited on Feb 25

Commit

30608fa

verified ·

1 Parent(s): 58bac6e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
-pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
-st.title("Hot Dog? Or Not?")
 file_name = st.file_uploader("Upload a hot dog candidate")
 if file_name is not None:
     col1, col2 = st.columns(2)

 from transformers import pipeline
 from PIL import Image
+pipeline = pipeline(task="image-classification", model="umm-maybe/AI-image-detector")
+st.title("AI Image Detector")
 file_name = st.file_uploader("Upload a hot dog candidate")
 if file_name is not None:
     col1, col2 = st.columns(2)