Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 from transformers import AutoProcessor, AutoModelForImageTextToText
 from PIL import Image
-from huggingface_hub import login
+import spaces
 
 # Load Vision-Instruct model
 processor = AutoProcessor.from_pretrained("alpindale/Llama-3.2-11B-Vision-Instruct")
 model = AutoModelForImageTextToText.from_pretrained("alpindale/Llama-3.2-11B-Vision-Instruct")
 
+@spaces.GPU
 def extract_text_from_image(image):
     """
     Function to extract text from a handwritten image using the Meta-Llama model.