seyma9gulsen commited on
Commit
15ef8ac
·
1 Parent(s): 767dbb5

model updated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from fastai.vision.all import *
2
  import gradio as gr
3
 
4
  # Load your trained model
5
- learn = load_learner("xception_deepfake_model.pkl")
6
 
7
  # Define the prediction function
8
  def predict_image(img):
 
2
  import gradio as gr
3
 
4
  # Load your trained model
5
+ learn = load_learner("deepfake_model.pkl")
6
 
7
  # Define the prediction function
8
  def predict_image(img):