Update app.py

app.py

@@ -2,40 +2,14 @@ import os
 import shutil
 import gradio as gr
 from transformers import ReactCodeAgent, HfEngine, Tool
-import pandas as pd
+import pandas as 
+import spaces
+import torch
 
 from gradio import Chatbot
 from streaming import stream_to_gradio
 from huggingface_hub import login
 from gradio.data_classes import FileData
-import torch
-
-
-# Initialize ZeroGPU
-if torch.cuda.is_available():
-    torch.backends.cudnn.benchmark = True
-    torch.backends.cudnn.enabled = True
-    torch.backends.cudnn.allow_tf32 = True
-
-    import os
-    os.environ['WANDB_DISABLED'] = 'true'
-    os.environ['TOKENIZERS_PARALLELIZM'] = 'false'
-    os.environ['TRANSFORMERS_PARALLELIZM'] = 'false'
-    os.environ['HF_REPO_REWRITE'] = 'true'
-
-    from zero_gpu import ZeroGpu
-    ZeroGpu(use_cpu=False)
-
-
-# Set HF_HOME
-os.environ['HF_HOME'] = 'C:/Users/jpkor/data/.huggingface'
-
-
-# Set HUGGINGFACE_HUB_CACHE
-os.environ['HUGGINGFACE_HUB_CACHE'] = 'C:/Users/jpkor/data/.huggingface/cache'
-
-
-login(os.getenv("HUGGINGFACEHUB_API_TOKEN"), add_to_git_credential=True)
 
 
 llm_engine = HfEngine("meta-llama/Meta-Llama-3.1-70B-Instruct")
@@ -79,6 +53,7 @@ Parent = mother, father
 Child = daughter, son, stepdaughter, stepson
 Some children travelled only with a nanny, therefore parch=0 for them."""
 
+@spaces.GPU
 def get_images_in_directory(directory):
     image_extensions = {'.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff'}
 
@@ -89,6 +64,7 @@ def get_images_in_directory(directory):
                 image_files.append(os.path.join(root, file))
     return image_files
 
+@spaces.GPU
 def interact_with_agent(file_input, additional_notes):
     shutil.rmtree("./figures")
     os.makedirs("./figures")