Update app.py

app.py

@@ -159,11 +159,11 @@ def update_task_status(category, task_id, status, result=None):
 
 @spaces.GPU(duration=150)
 def infer(request: gr.Request):
-    if request:
-        print("请求头字典:", request.headers)
-        print("IP 地址:", request.client.host)
-        print("查询参数:", dict(request.query_params))
-        print("会话哈希:", request.session_hash)
+    # if request:
+    #     print("请求头字典:", request.headers)
+    #     print("IP 地址:", request.client.host)
+    #     print("查询参数:", dict(request.query_params))
+    #     print("会话哈希:", request.session_hash)
 
     # Fetch tasks
     img2text_tasks = fetch_task('img2text', fetch_all=True)
@@ -179,7 +179,7 @@ def infer(request: gr.Request):
             print(prompt)
             # Fetch the image from the URL
             image_response = requests.get(image_url)
-            image = Image.open(BytesIO(image_response.content))
+            image = Image.open(BytesIO(image_response.content)).convert("RGB")
 
             # Resize image
             max_size = 256