Update app.py

app.py

@@ -5,24 +5,28 @@ import torch
 
 # Load model dan tokenizer
 model_id = "vikhyatk/moondream2"
-# Menggunakan revision spesifik untuk menghindari error 'all_tied_weights_keys'
+# Gunakan revision terbaru yang kompatibel dengan transformers 4.45+
 model = AutoModelForCausalLM.from_pretrained(
     model_id, 
     trust_remote_code=True, 
-    revision="2024-05-08"
+    revision="2024-08-26" # Update ke revisi yang lebih baru
 )
-tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-05-08")
+tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-08-26")
 
 def answer_question(image, question):
     if image is None:
         return "No image provided"
     
-    # Proses gambar ke format yang dikenali model
+    # Deteksi device (CPU/GPU)
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    model.to(device)
+    
+    # Proses gambar
     enc_image = model.encode_image(image)
     answer = model.answer_question(enc_image, question, tokenizer)
     return answer
 
-# Interface Gradio sederhana
+# Interface Gradio
 interface = gr.Interface(
     fn=answer_question,
     inputs=[gr.Image(type="pil"), gr.Textbox(label="Question")],