f

app.py

@@ -3,9 +3,10 @@ import pandas as pd
 from datasets import Dataset
 from transformers import AutoTokenizer, TrainingArguments, Trainer, AutoModelForCausalLM
 import torch
+import zerogpu
 
-print("CUDA available:", torch.cuda.is_available())
-print("Device:", torch.device('cpu'))
+# Initialize ZeroGPU
+zerogpu.initialize()
 
 def train_model(file, hf_token):
     try:

requirements.txt

@@ -4,3 +4,4 @@ torch==2.1.2
 datasets==2.16.1
 pandas==2.2.0
 huggingface-hub==0.27.0
+zerogpu