Update app.py

app.py

@@ -8,6 +8,7 @@ from transformers import BitsAndBytesConfig, pipeline
 import re
 import time
 
+Description = "YOLO"
 model_id = "llava-hf/llava-1.5-7b-hf"
 quantization_config = BitsAndBytesConfig(
     load_in_4bit=True,
@@ -41,6 +42,7 @@ def infer(image, prompt,
             min_length,
             top_p):
   print("类型是：",type(image))
+  print(temperature, length_penalty, repetition_penalty, max_length, min_length, top_p)
   outputs = pipe(images=image, prompt=prompt,
                   generate_kwargs={"temperature":temperature,
                   "length_penalty":length_penalty,