Upload 3 files

README.md

@@ -5,7 +5,7 @@ language:
 ---
 # Image Captioning App
 
-This is a mod of [Wi-zz/joy-caption-pre-alpha](https://huggingface.co/Wi-zz/joy-caption-pre-alpha) and [fancyfeast/joy-caption-alpha-two](https://huggingface.co/spaces/fancyfeast/joy-caption-alpha-two). Thanks to [dominic1021](https://huggingface.co/dominic1021), [IceHibiki](https://huggingface.co/IceHibiki), [BullseyeMxP](https://huggingface.co/BullseyeMxP).
+This is a mod of [Wi-zz/joy-caption-pre-alpha](https://huggingface.co/Wi-zz/joy-caption-pre-alpha) and [fancyfeast/joy-caption-alpha-two](https://huggingface.co/spaces/fancyfeast/joy-caption-alpha-two). Thanks to [dominic1021](https://huggingface.co/dominic1021), [IceHibiki](https://huggingface.co/IceHibiki), [BullseyeMxP](https://huggingface.co/BullseyeMxP), [Wakeme](https://huggingface.co/Wakeme).
 
 # Notice: I will contribute to Wi-zz after shaping the code.
 

app.py

@@ -21,9 +21,10 @@ IS_COLAB = 'google.colab' in sys.modules
 IMAGE_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.bmp', '.webp')
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 BASE_DIR = Path(__file__).resolve().parent # Define the base directory
-CHECKPOINT_PATH = BASE_DIR / Path("cgrkzexw-599808") if not IS_COLAB else Path("/content/joy-caption-alpha-two-cli-mod/cgrkzexw-599808")
+CHECKPOINT_PATH = BASE_DIR / Path("cgrkzexw-599808")
 CLIP_PATH = "google/siglip-so400m-patch14-384"
 DEFAULT_MODEL_PATH = "unsloth/Meta-Llama-3.1-8B-bnb-4bit"
+#DEFAULT_MODEL_PATH = "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit" # Default in Alpha One Two.
 #DEFAULT_MODEL_PATH = "Orenguteng/Llama-3.1-8B-Lexi-Uncensored-V2" # Works better but full weight.
 LORA_PATH = CHECKPOINT_PATH / "text_model"
 CAPTION_TYPE_MAP = {