Update app.py
Browse files
app.py
CHANGED
@@ -36,8 +36,8 @@ genai.configure(api_key="AIzaSyBGhEOy-JYMzGtTcRjBjP51OGR168WKRFw")
|
|
36 |
# return chat_completion.choices[0].message.content
|
37 |
|
38 |
|
39 |
-
@app.post("/upload-image/")
|
40 |
-
async def upload_image(file: UploadFile = File(...)):
|
41 |
os.makedirs("uploads", exist_ok=True)
|
42 |
|
43 |
|
@@ -49,7 +49,7 @@ async def upload_image(file: UploadFile = File(...)):
|
|
49 |
|
50 |
model = genai.GenerativeModel("gemini-1.5-pro-latest")
|
51 |
result = model.generate_content(
|
52 |
-
[myfile, "\n\n",
|
53 |
)
|
54 |
|
55 |
return result.text
|
|
|
36 |
# return chat_completion.choices[0].message.content
|
37 |
|
38 |
|
39 |
+
@app.post("/upload-image/{prompt}")
|
40 |
+
async def upload_image(file: UploadFile = File(...), prompt: str):
|
41 |
os.makedirs("uploads", exist_ok=True)
|
42 |
|
43 |
|
|
|
49 |
|
50 |
model = genai.GenerativeModel("gemini-1.5-pro-latest")
|
51 |
result = model.generate_content(
|
52 |
+
[myfile, "\n\n", prompt]
|
53 |
)
|
54 |
|
55 |
return result.text
|