Spaces:
Running
Running
rogerxavier
commited on
Commit
•
044ae9f
1
Parent(s):
db6c14b
Update app.py
Browse files
app.py
CHANGED
@@ -181,13 +181,12 @@ def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
|
|
181 |
|
182 |
|
183 |
#获取图片调用image_to_wd14_tags函数获取返回 ->"ratings, output_text, filtered_tags"
|
184 |
-
@app.
|
185 |
-
def getOriginalMangaList(image:
|
186 |
print("收到请求")
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
return image_to_wd14_tags(image_data,'chen-moat2',0.5,True,True)
|
191 |
|
192 |
|
193 |
if __name__ == "__main__":
|
|
|
181 |
|
182 |
|
183 |
#获取图片调用image_to_wd14_tags函数获取返回 ->"ratings, output_text, filtered_tags"
|
184 |
+
@app.post("/getOriginalMangaList")
|
185 |
+
def getOriginalMangaList(image: UploadFile = File(...)):
|
186 |
print("收到请求")
|
187 |
+
img = image.file.read()
|
188 |
+
image_data = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
|
189 |
+
return image_to_wd14_tags(image_data, 'chen-moat2', 0.5, True, True)
|
|
|
190 |
|
191 |
|
192 |
if __name__ == "__main__":
|