Update main.py

main.py

@@ -505,9 +505,7 @@ def generate_image(request: Request, input_data: InputImage):
 #--------------------------------------------------- IMAGE TO TEXT ------------------------------------------------------
 class InputImageToText(BaseModel): 
     base64: str
-    input_image: str = ''
-    input_text: str = '' 
-    max_new_tokens: int = 100
+    input: str = 'describe this image in detail'
 
 def base64_in_immagine(dati_base64):
     immagine = base64.b64decode(dati_base64)
@@ -521,7 +519,7 @@ def image_to_text(request: Request, input_data: InputImageToText):
     client = Client("vikhyatk/moondream2")
     result = client.predict(
     		"/tmp/img.jpg",	
-    		"describe this image in detail",	
+    		input_data.input,
     		api_name="/answer_question_1"
     )
     LoggaTesto("IMMAGINE", {"response": result}, False)