Update app.py

app.py

@@ -6,19 +6,6 @@ import torch
 from PIL import Image
 from transformers import AutoModel, AutoTokenizer
 
-
-import argparse
-from decord import VideoReader, cpu
-import io
-import os
-import copy
-import requests
-import base64
-import json
-import traceback
-import re
-import modelscope_studio as mgr
-
 # Pre-Initialize
 DEVICE = "auto"
 if DEVICE == "auto":
@@ -31,7 +18,6 @@ DEFAULT_INPUT = "Describe in one paragraph."
 repo = AutoModel.from_pretrained("openbmb/MiniCPM-V-2_6", torch_dtype=torch.bfloat16, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("openbmb/MiniCPM-V-2_6", trust_remote_code=True)
 
-repo.eval()
 css = '''
 .gradio-container{max-width: 560px !important}
 h1{text-align:center}
@@ -46,10 +32,10 @@ def generate(image, instruction=DEFAULT_INPUT, sampling=False, temperature=0.7,
     global model, tokenizer
 
     print(image)
-    image_rgb = Image.open(image).convert("RGB")
-    print(image_rgb, instruction)
+    # image_rgb = Image.open(image).convert("RGB")
+    # print(image_rgb, instruction)
 
-    inputs = [{"role": "user", "content": [image_rgb, instruction]}]
+    inputs = [{"role": "user", "content": [image, instruction]}]
 
     parameters = {
         "sampling": sampling,