spaces added

app.py

@@ -11,6 +11,7 @@ import os
 from pathlib import Path
 import tempfile
 import torch
+import spaces
 
 # Global variables to store the loaded model and tokenizer
 dental_model = None
@@ -44,6 +45,8 @@ def load_dental_transformers_model():
             return None, None
     return dental_model, dental_tokenizer
 
+
+@spaces.GPU(duration=120)
 def generate_dental_response(
     question: str, 
     max_tokens: int = 2048, 
@@ -465,7 +468,7 @@ if __name__ == "__main__":
     demo.launch(
             server_name="0.0.0.0",
             server_port=7860,
-            share=True,
+            share=False,
             show_error=True,
             ssr_mode=False  # Disable SSR for Spaces compatibility
         )

requirements.txt

@@ -5,4 +5,5 @@ langextract==1.0.3
 pandas
 numpy
 requests
-accelerate
+accelerate
+spaces