Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -509,8 +509,9 @@ class InputImageToText(BaseModel):
|
|
509 |
|
510 |
def base64_in_immagine(dati_base64):
|
511 |
immagine = base64.b64decode(dati_base64)
|
|
|
512 |
immagine_pil = Image.open(BytesIO(immagine))
|
513 |
-
nome_file = "img.jpg"
|
514 |
immagine_pil.save(nome_file)
|
515 |
|
516 |
@app.post("/Image_To_Text")
|
@@ -519,7 +520,7 @@ def image_to_text(request: Request, input_data: InputImageToText):
|
|
519 |
print(input_data.base64)
|
520 |
client = Client("vikhyatk/moondream2")
|
521 |
result = client.predict(
|
522 |
-
"img.jpg",
|
523 |
"",
|
524 |
api_name="/answer_question_1"
|
525 |
)
|
|
|
509 |
|
510 |
def base64_in_immagine(dati_base64):
|
511 |
immagine = base64.b64decode(dati_base64)
|
512 |
+
f"/tmp/image_{os.getpid()}.png"
|
513 |
immagine_pil = Image.open(BytesIO(immagine))
|
514 |
+
nome_file = "/tmp/img.jpg"
|
515 |
immagine_pil.save(nome_file)
|
516 |
|
517 |
@app.post("/Image_To_Text")
|
|
|
520 |
print(input_data.base64)
|
521 |
client = Client("vikhyatk/moondream2")
|
522 |
result = client.predict(
|
523 |
+
"/tmp/img.jpg",
|
524 |
"",
|
525 |
api_name="/answer_question_1"
|
526 |
)
|