Add application file

app.py

@@ -2,6 +2,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import gradio as gr
 from peft import PeftModel
+import spaces
 
 # Define the base model ID
 base_model_id = "meta-llama/Llama-2-13b-hf"
@@ -34,6 +35,7 @@ def formatting_func(job_description):
     text = f"### The job description: {job_description}\n ### The skills: "
     return text
 
+@spaces.GPU
 def generate_skills(job_description):
     formatted_text = formatting_func(job_description)
     model_input = tokenizer(formatted_text, return_tensors="pt").to("cuda")