Update app.py

app.py

@@ -9,11 +9,11 @@ import os
 
 
 DESCRIPTION = """
-# SmolVLM-trl-sft-ChartQA Demo
+# SmolVLM-trl-dpo-rlaif-v Demo
 
-This is a demo Space for a fine-tuned version of [SmolVLM](https://huggingface.co/HuggingFaceTB/SmolVLM-Instruct) trained using [ChatQA dataset](https://huggingface.co/datasets/HuggingFaceM4/ChartQA).
+This is a demo Space for a fine-tuned version of [SmolVLM](https://huggingface.co/HuggingFaceTB/SmolVLM-Instruct) trained using [rlaif-v dataset](https://huggingface.co/datasets/HuggingFaceH4/rlaif-v_formatted).
 
-The corresponding model is located [here](https://huggingface.co/sergiopaniego/smolvlm-instruct-trl-sft-ChartQA).
+The corresponding model is located [here](https://huggingface.co/sergiopaniego/smolvlm-instruct-trl-dpo-rlaif-v).
 """
 
 model_id = "HuggingFaceTB/SmolVLM-Instruct"
@@ -25,7 +25,7 @@ model = Idefics3ForConditionalGeneration.from_pretrained(
 )
 
 processor = AutoProcessor.from_pretrained(model_id)
-adapter_path = "sergiopaniego/smolvlm-instruct-trl-sft-ChartQA"
+adapter_path = "sergiopaniego/smolvlm-instruct-trl-dpo-rlaif-v"
 model.load_adapter(adapter_path)
 
 def array_to_image_path(image_array):
@@ -102,7 +102,7 @@ css = """
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(DESCRIPTION)
-    with gr.Tab(label="SmolVLM-trl-sft-ChartQA Input"):
+    with gr.Tab(label="SmolVLM-trl-dpo-rlaif-v Input"):
         with gr.Row():
             with gr.Column():
                 input_img = gr.Image(label="Input Picture")