调用llama3.2-vision,自动识别图片，输出prompt

2025-06-10 17:41:32 +08:00
parent cde7fe09ee
commit 72f24c9d14
4 changed files with 78 additions and 2 deletions
--- a/app/schemas/prompt_generation.py
+++ b/app/schemas/prompt_generation.py
@@ -3,3 +3,7 @@ from pydantic import BaseModel

 class PromptGenerationImageModel(BaseModel):
    text: str
+
+
+class ImageRequest(BaseModel):
+    img: str