Update app.py
Browse files
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(
|
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(
|
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)
|