Spaces:

ReacherTN
/

ai_image_gen_detector

Runtime error

ReacherTN commited on Apr 21, 2023

Commit

8693834

1 Parent(s): 60d40f7

Update predict.py

Files changed (1) hide show

predict.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import torch.nn.functional as F
 from model import CustomModel
-CKPT = 'fold_0.pt'
 Targets = ['Not AI' 'AI Generated']
 def predict_one_image(path) :
     image = read_image(path)

 import torch.nn.functional as F
 from model import CustomModel
+CKPT = 'fold_0'
 Targets = ['Not AI' 'AI Generated']
 def predict_one_image(path) :
     image = read_image(path)