feat: markdown

TinyLLaVA_Factory/tinyllava/serve/app.py

@@ -21,7 +21,7 @@ from tinyllava.utils import *
 from tinyllava.data import *
 from tinyllava.model import *
 
-DEFAULT_MODEL_PATH = "tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B"
+DEFAULT_MODEL_PATH = "cpu4dream/llava-small-OpenELM-AIMv2-0.6B"
 
 
 block_css = """
@@ -31,8 +31,11 @@ block_css = """
 }
 """
 title_markdown = """
-# TinyLLaVA: A Framework of Small-scale Large Multimodal Models
-[[Code](https://github.com/DLCV-BUAA/TinyLLaVABench)] | 📚 [[Paper](https://arxiv.org/pdf/2402.14289.pdf)]
+# Tiny Llava OpenELM-AIMv2 0.6B 🐛
+
+## Multimodal Image Question Answering on CPU
+
+This space demonstrates the capabilities of the [cpu4dream/llava-small-OpenELM-AIMv2-0.6B](https://huggingface.co/cpu4dream/llava-small-OpenELM-AIMv2-0.6B) model, trained using the [TinyLLaVA Framework](https://github.com/TinyLLaVA/TinyLLaVA_Factory).
 """
 tos_markdown = """
 ### Terms of use
@@ -345,6 +348,7 @@ if __name__ == "__main__":
     logger.info(gr.__version__)
     args = parse_args()
     model_name = args.model_name
+    DEFAULT_MODEL_PATH = args.model_path
     model, tokenizer, image_processor, context_len = load_pretrained_model(
         args.model_path,
         load_4bit=args.load_4bit,