Upload app.py with huggingface_hub

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from zerogpu import setup_zerogpu
 
 # Load model and tokenizer
 model_id = "LiquidAI/LFM2-2.6B"
+setup_zerogpu()
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
-    device_map="auto",
-    torch_dtype="bfloat16",
+    device_map="auto",  # Ensure proper device mapping for zero-gpu
 )
 tokenizer = AutoTokenizer.from_pretrained(model_id)