Update app.py

app.py

@@ -2,7 +2,7 @@ from transformers import AutoConfig, AutoModel
 import gradio as gr
 import spaces
 
-model_path = "turing-motors/Heron-NVILA-Lite-33B"
+model_path = "turing-motors/Heron-NVILA-Lite-15B"
 
 # or directly from_pretrained
 model = AutoModel.from_pretrained(model_path, trust_remote_code=True, device_map="auto")
@@ -20,13 +20,13 @@ generation_config = GenerationConfig(**generation_config)
 
 DESCRIPTION = '''
 <div>
-<h1 style="text-align: center;">非公式Heron-NVILA-Lite</h1>
-<p>Heron-NVILA-Liteの非公式デモだよ。 <a href="https://huggingface.co/turing-motors/Heron-NVILA-Lite-33B"><b>turing-motors/Heron-NVILA-Lite-33B</b></a>.</p>
+<h1 style="text-align: center;">非公式Heron-NVILA-Lite-15B</h1>
+<p>Heron-NVILA-Lite-15Bの非公式デモだよ。 <a href="https://huggingface.co/turing-motors/Heron-NVILA-Lite-15B"><b>turing-motors/Heron-NVILA-Lite-15B</b></a>.</p>
 </div>
 '''
 
 @spaces.GPU()
-def infer(image,prompt):
+def infer(image):
     response = model.generate_content(
         [image, "画像を可能な限り詳細に説明してください。"],
         generation_config=generation_config