Update app.py

app.py

@@ -2,12 +2,14 @@ import streamlit as st
 from llama_cpp import Llama
 from PIL import Image
 import numpy as np
+from transformers import AutoModel
+model = AutoModel.from_pretrained("liminerity/bitmap-mistral-M7-slerp-alpaca-70m-gguf")
 
 # Initialize the model (only once)
 @st.cache_resource
 def load_model():
     return Llama(
-        model_path="models/bitmap-mistral-M7-slerp-alpaca-70m.gguf",
+        model_path=model,
         n_ctx=2048,
         n_threads=4
     )
@@ -22,7 +24,7 @@ if st.button("Generate Bitmap"):
         # Generate the bitmap description
         llm = load_model()
         response = llm(
-            f"Generate a 32x32 bitmap array using 0s and 1s that represents {prompt}. "
+            f"Generate a 64x64 bitmap array using 0s and 1s that represents {prompt}. "
             "Only output the array, no other text.",
             max_tokens=2048,
             temperature=0.7