krammnic commited on
Commit
daca96e
·
verified ·
1 Parent(s): 34b86c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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