nyasukun commited on
Commit
f94978f
·
verified ·
1 Parent(s): 9b70e7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -102,7 +102,7 @@ class LocalModelManager:
102
  logger.error(f"Error loading model {model_path}: {str(e)}")
103
  raise
104
 
105
- @spaces.GPU(duration=120)
106
  def _generate_text_sync(self, pipeline, text: str) -> str:
107
  """同期的なテキスト生成の実行"""
108
  outputs = pipeline(
@@ -126,7 +126,7 @@ class LocalModelManager:
126
  logger.error(f"Error in text generation with {model_path}: {str(e)}")
127
  raise
128
 
129
- @spaces.GPU(duration=60)
130
  def _classify_text_sync(self, pipeline, text: str) -> str:
131
  """同期的なテキスト分類の実行"""
132
  result = pipeline(text)
 
102
  logger.error(f"Error loading model {model_path}: {str(e)}")
103
  raise
104
 
105
+ @spaces.GPU()
106
  def _generate_text_sync(self, pipeline, text: str) -> str:
107
  """同期的なテキスト生成の実行"""
108
  outputs = pipeline(
 
126
  logger.error(f"Error in text generation with {model_path}: {str(e)}")
127
  raise
128
 
129
+ @spaces.GPU()
130
  def _classify_text_sync(self, pipeline, text: str) -> str:
131
  """同期的なテキスト分類の実行"""
132
  result = pipeline(text)