Update app.py

app.py

@@ -21,14 +21,10 @@ moondream.eval()
 @spaces.GPU
 def answer_questions(image_tuples, prompt_text):
     result = ""
-    
+    Q_and_A = ""
     prompts = [p.strip() for p in prompt_text.split(',')]  # Splitting and cleaning prompts
-    print(f"prompts\n{prompts}\n")
+    print(f"prompts: {prompts}\n")
     image_embeds = [img[0] for img in image_tuples if img[0] is not None]  # Extracting images from tuples, ignoring None
-    
-    # Check if the lengths of image_embeds and prompts are equal
-    #if len(image_embeds) != len(prompts):
-        #return ("Error: The number of images input and prompts input (seperate by commas in input text field) must be the same.")
 
     answers = []
     for prompt in prompts:
@@ -51,19 +47,7 @@ def answer_questions(image_tuples, prompt_text):
     print(f"\n\n{Q_and_A}\n\n")
     
     result = {'headers': prompts, 'data': data}
-    return result
-    '''
-    answers = moondream.batch_answer(
-        images=image_embeds,
-        prompts=prompts,
-        tokenizer=tokenizer,
-    )
-    
-    for question, answer in zip(prompts, answers):
-        result += (f"Q: {question}\nA: {answer}\n\n")
-        
-    return result
-    '''
+    return Q_and_A, result
 
 with gr.Blocks() as demo:
     gr.Markdown("# moondream2 unofficial batch processing demo")