add space

medbot/handlers.py

@@ -1,3 +1,4 @@
+import spaces
 from .model import ModelManager
 from .memory import MedicalMemoryManager
 from .prompts import CONSULTATION_PROMPT, MEDICINE_PROMPT
@@ -17,6 +18,7 @@ def build_me_llama_prompt(system_prompt, history, user_input):
     prompt += f"{user_input} [/INST] "
     return prompt
 
+@spaces.GPU
 def respond(message, chat_history):
     global conversation_turns
     conversation_turns += 1