Update app.py

app.py

@@ -11,6 +11,7 @@ from transformers import (
     AutoModelForCausalLM)
 
 
+@spaces.GPU
 def generate_prompt(instruction, input=None):
     if input:
         return f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.  # noqa: E501
@@ -33,6 +34,7 @@ def generate_prompt(instruction, input=None):
 """
 
 
+@spaces.GPU
 based_model_path = "meta-llama/Meta-Llama-3-8B"
 lora_weights = "NouRed/BioMed-Tuned-Llama-3-8b"