Update app.py

app.py

@@ -5,33 +5,40 @@ import torch
 
 # Load model dan tokenizer
 model_id = "vikhyatk/moondream2"
-# Gunakan revision terbaru yang kompatibel dengan transformers 4.45+
+revision = "2024-08-26"
+
+# Deteksi device (Gunakan GPU jika tersedia, jika tidak gunakan CPU)
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 model = AutoModelForCausalLM.from_pretrained(
     model_id, 
     trust_remote_code=True, 
-    revision="2024-08-26" # Update ke revisi yang lebih baru
-)
-tokenizer = AutoTokenizer.from_pretrained(model_id, revision="2024-08-26")
+    revision=revision
+).to(device)
+tokenizer = AutoTokenizer.from_pretrained(model_id, revision=revision)
 
 def answer_question(image, question):
     if image is None:
         return "No image provided"
     
-    # Deteksi device (CPU/GPU)
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    model.to(device)
+    # Pastikan gambar dalam format RGB
+    image = image.convert("RGB")
+    
+    # Proses gambar dan dapatkan jawaban
+    with torch.no_grad():
+        enc_image = model.encode_image(image)
+        answer = model.answer_question(enc_image, question, tokenizer)
     
-    # Proses gambar
-    enc_image = model.encode_image(image)
-    answer = model.answer_question(enc_image, question, tokenizer)
     return answer
 
 # Interface Gradio
 interface = gr.Interface(
     fn=answer_question,
-    inputs=[gr.Image(type="pil"), gr.Textbox(label="Question")],
+    inputs=[gr.Image(type="pil"), gr.Textbox(label="Question", placeholder="What is written in the captcha?")],
     outputs=gr.Text(label="Answer"),
-    title="Moondream Captcha Solver"
+    title="Moondream Captcha Solver",
+    description="Upload image captcha dan biarkan model menjawabnya."
 )
 
-interface.launch()
+if __name__ == "__main__":
+    interface.launch()