Update main.py

main.py

@@ -505,7 +505,9 @@ def generate_image(request: Request, input_data: InputImage):
 #--------------------------------------------------- IMAGE TO TEXT ------------------------------------------------------
 class InputImageToText(BaseModel): 
     base64: str
-    input: str = ''
+    input_image: str = ''
+    input_text: str = '' 
+    max_new_tokens: int = 100
 
 def base64_in_immagine(dati_base64):
     immagine = base64.b64decode(dati_base64)
@@ -516,14 +518,13 @@ def base64_in_immagine(dati_base64):
 @app.post("/Image_To_Text")
 def image_to_text(request: Request, input_data: InputImageToText):
     base64_in_immagine(input_data.base64)
-    print(input_data.base64)
     client = Client("vikhyatk/moondream2")
     result = client.predict(
     		"/tmp/img.jpg",	
     		"",	
     		api_name="/answer_question_1"
     )
-    print(result)
+    LoggaTesto("DESCRIZIONE IMMAGINE", {"response": result}, False)
     return {"response": result}
     
 #--------------------------------------------------- API PostSpazio ------------------------------------------------------