Add title and descriptions

app.py

@@ -4,7 +4,7 @@ from prompt import smoke_detection_prompt
 import gradio as gr
 import spaces
 
-model_name = "leon-se/ForestFireVLM-3B"
+model_name = "leon-se/ForestFireVLM-7B"
 model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     model_name, torch_dtype="auto", device_map="auto"
 )
@@ -40,7 +40,7 @@ def generate(image):
     inputs = inputs.to("cuda")
 
     # Inference: Generation of the output
-    generated_ids = model.generate(**inputs, max_new_tokens=300)
+    generated_ids = model.generate(**inputs, max_new_tokens=300, do_sample=False)
     generated_ids_trimmed = [
         out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
     ]
@@ -52,5 +52,8 @@ def generate(image):
 inputs = gr.Image(type="pil", label="Input Image")
 outputs = gr.JSON(label="Output")
 
+title = "ForestFireVLM-7B"
+description = "This is a ForestFireVLM-7B, a finetune of Qwen2.5-VL-7B-Instruct. This demo shows how Vision-Language Models can give detailled and structured captions for forest fires from UAV perspectives."
+
 demo = gr.Interface(fn=generate, inputs=inputs, outputs=outputs, deep_link=False)
 demo.launch()