Spaces:
Sleeping
Sleeping
Commit
·
e5d9d3a
1
Parent(s):
4f01bac
test
Browse files
app.py
CHANGED
|
@@ -4,6 +4,10 @@ import re
|
|
| 4 |
import textstat
|
| 5 |
from transformers import pipeline
|
| 6 |
from langdetect import detect
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# Load a summarization model
|
| 9 |
summarizer = pipeline("summarization")
|
|
|
|
| 4 |
import textstat
|
| 5 |
from transformers import pipeline
|
| 6 |
from langdetect import detect
|
| 7 |
+
import os
|
| 8 |
+
|
| 9 |
+
# Disable GPU usage for TensorFlow
|
| 10 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
|
| 11 |
|
| 12 |
# Load a summarization model
|
| 13 |
summarizer = pipeline("summarization")
|