vicliv commited on
Commit
72f04b1
·
1 Parent(s): e1f0077

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -166,12 +166,12 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
166
 
167
  sorted_scores = sorted(leaderboard_scores.items(), key=lambda x: x[1], reverse=True)
168
  leaderboard_table = [[name, points] for name, points in sorted_scores]
169
-
170
  type_image = "real" if model_name.lower() == "real" else "fake"
171
  image_dir = os.path.join("test", type_image)
172
  os.makedirs(image_dir, exist_ok=True)
173
- image_id = random.randint(10000000, 99999999)
174
- image_filename = f"{image_id}.jpg"
175
  image_path = os.path.join(image_dir, image_filename)
176
  image.save(image_path)
177
 
 
166
 
167
  sorted_scores = sorted(leaderboard_scores.items(), key=lambda x: x[1], reverse=True)
168
  leaderboard_table = [[name, points] for name, points in sorted_scores]
169
+
170
  type_image = "real" if model_name.lower() == "real" else "fake"
171
  image_dir = os.path.join("test", type_image)
172
  os.makedirs(image_dir, exist_ok=True)
173
+ timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
174
+ image_filename = f"{timestamp}.jpg"
175
  image_path = os.path.join(image_dir, image_filename)
176
  image.save(image_path)
177