rein0421 commited on
Commit
407a5d0
·
verified ·
1 Parent(s): 146e98d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -59,7 +59,7 @@ import cv2
59
  import numpy as np
60
  from ultralytics import YOLO # YOLOv8ライブラリ
61
 
62
-
63
  import cv2
64
  import numpy as np
65
  from datetime import datetime
@@ -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
-
741
 
742
  @app.post("/create-mask-and-inpaint-simple-lama-special")
743
  async def create_mask_and_inpaint_simple_lama(
 
59
  import numpy as np
60
  from ultralytics import YOLO # YOLOv8ライブラリ
61
 
62
+ import random
63
  import cv2
64
  import numpy as np
65
  from datetime import datetime
 
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_number = random.randint(1, 100)#これはバグってるときのデモ用、本番はコメントアウトして
741
 
742
  @app.post("/create-mask-and-inpaint-simple-lama-special")
743
  async def create_mask_and_inpaint_simple_lama(