litav commited on
Commit
6917b4f
·
verified ·
1 Parent(s): b819af8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -10,13 +10,20 @@ def predict(image: Image.Image):
10
  label, confidence = model.predict(image)
11
  result = "AI image" if label == 1 else "Real image"
12
  return result, f"Confidence: {confidence:.2f}%"
 
 
 
 
 
 
 
 
13
 
14
 
15
  # Define the Gradio interface
16
  demo = gr.Interface(
17
  fn=predict,
18
  inputs=gr.Image(type="pil"),
19
- gradio.PlayableVideo("https://cdn-uploads.huggingface.co/production/uploads/66d6f1b3b50e35e1709bfdf7/x7Ud8PO9QPfmrTvBVcCKE.mp4"),
20
  outputs=[gr.Textbox(), gr.Textbox()],
21
  title="Vision Transformer Model",
22
  description="Upload an image to classify it using the Vision Transformer model."
 
10
  label, confidence = model.predict(image)
11
  result = "AI image" if label == 1 else "Real image"
12
  return result, f"Confidence: {confidence:.2f}%"
13
+
14
+ def video_identity(video):
15
+ return video
16
+
17
+ demo = gr.Interface(video_identity,
18
+ gr.Video(),
19
+ "https://cdn-uploads.huggingface.co/production/uploads/66d6f1b3b50e35e1709bfdf7/x7Ud8PO9QPfmrTvBVcCKE.mp4",
20
+ )
21
 
22
 
23
  # Define the Gradio interface
24
  demo = gr.Interface(
25
  fn=predict,
26
  inputs=gr.Image(type="pil"),
 
27
  outputs=[gr.Textbox(), gr.Textbox()],
28
  title="Vision Transformer Model",
29
  description="Upload an image to classify it using the Vision Transformer model."