vicliv commited on
Commit
e1f0077
·
1 Parent(s): 87d68a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -176,7 +176,7 @@ def detect_with_model(image: Image.Image, prompt: str, username: str, model_name
176
  image.save(image_path)
177
 
178
  csv_path = os.path.join("test", "leaderboard_entries.csv")
179
- header = "file_name,prompt,label,model,split,user,timestamp\n"
180
  csv_line = f"test/{type_image}/{image_filename},\"{prompt}\",{type_image},{model_name},test,{prediction.lower()},{username},{datetime.now().isoformat()}\n"
181
  try:
182
  file_exists = os.path.exists(csv_path)
 
176
  image.save(image_path)
177
 
178
  csv_path = os.path.join("test", "leaderboard_entries.csv")
179
+ header = "file_name,prompt,label,model,split,prediction,user,timestamp\n"
180
  csv_line = f"test/{type_image}/{image_filename},\"{prompt}\",{type_image},{model_name},test,{prediction.lower()},{username},{datetime.now().isoformat()}\n"
181
  try:
182
  file_exists = os.path.exists(csv_path)