wuhp commited on
Commit
a85a496
·
verified ·
1 Parent(s): a4166cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -3
app.py CHANGED
@@ -9,7 +9,7 @@ import cv2
9
  import numpy as np
10
  from PIL import Image
11
  import gradio as gr
12
- from roboflow import Roboflow # removed RoboflowError, just import Roboflow
13
 
14
 
15
  def parse_roboflow_url(url: str):
@@ -159,7 +159,7 @@ def upload_and_train_detection(
159
  rf = Roboflow(api_key=api_key)
160
  ws = rf.workspace(workspace)
161
 
162
- # get‑or‑create project by inspecting exception text
163
  try:
164
  proj = ws.project(project_slug)
165
  except Exception as e:
@@ -173,7 +173,18 @@ def upload_and_train_detection(
173
  else:
174
  raise
175
 
176
- # upload & train
 
 
 
 
 
 
 
 
 
 
 
177
  ws.upload_dataset(
178
  dataset_path,
179
  project_slug,
@@ -181,6 +192,7 @@ def upload_and_train_detection(
181
  project_type=project_type
182
  )
183
 
 
184
  version_num = proj.generate_version(settings={
185
  "augmentation": {},
186
  "preprocessing": {},
 
9
  import numpy as np
10
  from PIL import Image
11
  import gradio as gr
12
+ from roboflow import Roboflow
13
 
14
 
15
  def parse_roboflow_url(url: str):
 
159
  rf = Roboflow(api_key=api_key)
160
  ws = rf.workspace(workspace)
161
 
162
+ # 1) Try to fetch existing project
163
  try:
164
  proj = ws.project(project_slug)
165
  except Exception as e:
 
173
  else:
174
  raise
175
 
176
+ # 2) If it exists but is NOT object-detection, make a fresh slug
177
+ if getattr(proj, "annotation", None) != project_type:
178
+ new_slug = project_slug + "-v2"
179
+ proj = ws.create_project(
180
+ new_slug,
181
+ annotation=project_type,
182
+ project_type=project_type,
183
+ project_license=project_license
184
+ )
185
+ project_slug = new_slug
186
+
187
+ # 3) Upload train/val/test
188
  ws.upload_dataset(
189
  dataset_path,
190
  project_slug,
 
192
  project_type=project_type
193
  )
194
 
195
+ # 4) Generate new version & train
196
  version_num = proj.generate_version(settings={
197
  "augmentation": {},
198
  "preprocessing": {},