Update app.py

app.py

@@ -170,49 +170,43 @@ def infer(request: gr.Request):
         return "No tasks found or failed to fetch tasks."
 
     for task in img2text_tasks:
-        if task['status'] == 'Pending':
-            try:
-                image_url = task['content']['url']
-                prompt = task['content']['prompt']
-
-                # Fetch the image from the URL
-                image_response = requests.get(image_url)
-                image = Image.open(BytesIO(image_response.content))
-
-                # Resize image
-                max_size = 256
-                width, height = image.size
-                if width > height:
-                    new_width = max_size
-                    new_height = int((new_width / width) * height)
-                else:
-                    new_height = max_size
-                    new_width = int((new_height / height) * width)
-                image = image.resize((new_width, new_height), Image.LANCZOS)
-
-                # Process the image and prompt
-                inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
-                generated_ids = model.generate(
-                    input_ids=inputs["input_ids"],
-                    pixel_values=inputs["pixel_values"],
-                    max_new_tokens=1024,
-                    do_sample=False,
-                    num_beams=3
-                )
-
-                generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
-                parsed_answer = processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
-                print(task['id'])
-                print(parsed_answer)
-
-                # Update the task status to Successed with result
-                update_task_status('img2text', task['id'], 'Successed', {"text": parsed_answer})
-
-            except Exception as e:
-                print(f"Error processing task {task['id']}: {e}")
-                # If error occurs, update the task status to Failed
-                update_task_status('img2text', task['id'], 'Failed')
-                return f"Error processing task {task['id']}: {e}"
+        try:
+            image_url = task['content']['url']
+            prompt = task['content']['prompt']
+            print(image_url)
+            print(prompt)
+            # Fetch the image from the URL
+            image_response = requests.get(image_url)
+            image = Image.open(BytesIO(image_response.content))
+
+            # Resize image
+            max_size = 256
+            width, height = image.size
+            if width > height:
+                new_width = max_size
+                new_height = int((new_width / width) * height)
+            else:
+                new_height = max_size
+                new_width = int((new_height / height) * width)
+            image = image.resize((new_width, new_height), Image.LANCZOS)
+
+            # Process the image and prompt
+            inputs = processor(text=prompt, images=image, return_tensors="pt").to(device)
+            generated_ids = model.generate(
+                input_ids=inputs["input_ids"],
+                pixel_values=inputs["pixel_values"],
+                max_new_tokens=1024,
+                do_sample=False,
+                num_beams=3
+            )
+
+            generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
+            parsed_answer = processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
+            print(task['id'])
+            print(parsed_answer)
+
+            # Update the task status to Successed with result
+            update_task_status('img2text', task['id'], 'Successed', {"text": parsed_answer})
 
     return "No pending tasks found."