Update main.py

main.py

@@ -516,12 +516,21 @@ 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)
-    client = Client("vikhyatk/moondream2")
-    result = client.predict(
-    		"/tmp/img.jpg",	
-    		input_data.input,
-    		api_name="/answer_question_1"
-    )
+    Version = 1
+    if Version == 1:
+        client = Client("https://vikhyatk-moondream1.hf.space/--replicas/av7ct/")
+        result = client.predict(
+                "/tmp/img.jpg",	
+        		input_data.input,
+        		api_name="/answer_question"
+        )
+    else: 
+        client = Client("vikhyatk/moondream2")
+        result = client.predict(
+        		"/tmp/img.jpg",	
+        		input_data.input,
+        		api_name="/answer_question_1"
+        )
     LoggaTesto("IMMAGINE", {"response": result}, False)
     return {"response": result}