Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 import spaces  # Add spaces import for Hugging Face Spaces
 import os
 
-hf_token = os.environ.get("API_KEY")
+hf_token = os.getenv("API_KEY")
 
 # Model information
 MODEL_ID = "DeepMount00/SmolVLM-Base-ocr_base"
@@ -49,7 +49,7 @@ def process_image(image, progress=gr.Progress()):
     progress(0.4, desc="Processing with model...")
     prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
     inputs = processor(text=prompt, images=[image], return_tensors="pt")
-    inputs = inputs.to('cuda')  # Move inputs to CUDA
+    inputs = inputs.to('cuda')
 
     # Generate outputs
     progress(0.6, desc="Generating text...")