Jay-Rajput commited on
Commit
89ed570
·
1 Parent(s): 30c60ea

ai detector caching

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -10,9 +10,10 @@ import numpy as np
10
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
11
  import time
12
  import json
 
13
 
14
- # Initialize models (simplified for Spaces deployment)
15
- @gr.Interface.cache
16
  def load_models():
17
  """Load lightweight models for Hugging Face Spaces"""
18
  try:
 
10
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
11
  import time
12
  import json
13
+ import functools
14
 
15
+ # Initialize models
16
+ @functools.lru_cache(maxsize=1)
17
  def load_models():
18
  """Load lightweight models for Hugging Face Spaces"""
19
  try: