alkatraz445 commited on
Commit
e27f696
·
verified ·
1 Parent(s): d2c18fe

deleted cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ for example_name in examples_names:
21
  np.random.shuffle(examples)
22
 
23
  # Load the Hugging Face model
24
- model = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection", device="cuda")
25
 
26
  # Define the prediction function
27
  def classify_image(image):
 
21
  np.random.shuffle(examples)
22
 
23
  # Load the Hugging Face model
24
+ model = pipeline("image-classification", model="dima806/deepfake_vs_real_image_detection")
25
 
26
  # Define the prediction function
27
  def classify_image(image):