Update app.py

app.py

@@ -1,15 +1,12 @@
-import os
-import torch
-import requests
-from io import BytesIO
-from PIL import Image, ImageSequence
+import os, torch
 from transformers import AutoProcessor, LlavaForConditionalGeneration
 import gradio as gr
+from PIL import Image, ImageSequence
+import requests
+from io import BytesIO
 
-# -------------------------------------------------
-# Model configuration (CPU‑only)
-# -------------------------------------------------
-MODEL_NAME = "fpgaminer/joycaption-llama3.1-8b"
+# ---- 1️⃣ Use a public repo ----
+MODEL_NAME = "llava-hf/joycaption-llama3.1-8b"   # public version
 
 processor = AutoProcessor.from_pretrained(MODEL_NAME)
 llava_model = LlavaForConditionalGeneration.from_pretrained(