rein0421 commited on
Commit
be0d45b
·
verified ·
1 Parent(s): d00caed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -736,8 +736,8 @@ async def classify_image(file: UploadFile = File(...)):
736
  # 新しい画像がどのクラスタに分類されるかを判定
737
  new_image_cluster = classify_new_image(new_image_vector, sums_data, loaded_vectors, loaded_object_names)
738
 
739
- #return {"danger":dangerarray[int(new_image_cluster + 1)]}#バグったらここを+にして
740
- return random.randint(1, 100)#これはバグってるときのデモ用、本番はコメントアウトして
741
 
742
  @app.post("/create-mask-and-inpaint-simple-lama-special")
743
  async def create_mask_and_inpaint_simple_lama(
 
736
  # 新しい画像がどのクラスタに分類されるかを判定
737
  new_image_cluster = classify_new_image(new_image_vector, sums_data, loaded_vectors, loaded_object_names)
738
 
739
+ return {"danger":dangerarray[int(new_image_cluster + 1)]}#バグったらここを+にして
740
+
741
 
742
  @app.post("/create-mask-and-inpaint-simple-lama-special")
743
  async def create_mask_and_inpaint_simple_lama(