Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,6 @@ tokenizer, generator_conf, model = load_model(REPO_NAME)
|
|
21 |
|
22 |
global_error = ''
|
23 |
try:
|
24 |
-
@spaces.GPU()
|
25 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
26 |
except Exception as e:
|
27 |
global_error = f"Failed to load model: {str(e)}"
|
|
|
21 |
|
22 |
global_error = ''
|
23 |
try:
|
|
|
24 |
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
25 |
except Exception as e:
|
26 |
global_error = f"Failed to load model: {str(e)}"
|