Hugging Face's logo

Spaces:
somosnlp
/
NoticIA-demo
Runtime error

App Files Community
Iker commited on
Commit
636adf6
路
verified 路
1 Parent(s): 12ea223

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -36,6 +36,8 @@ model = AutoModelForCausalLM.from_pretrained(
36
  quantization_config=quantization_config,
37
  )
38
 
 
 
39
  # Par谩metros de generaci贸n.
40
  generation_config = GenerationConfig(
41
  max_new_tokens=128, # Los res煤menes son cortos, no necesitamos m谩s tokens
 
36
  quantization_config=quantization_config,
37
  )
38
 
39
+ print(f"Model loaded in {model.device}")
40
+
41
  # Par谩metros de generaci贸n.
42
  generation_config = GenerationConfig(
43
  max_new_tokens=128, # Los res煤menes son cortos, no necesitamos m谩s tokens
requirements.txt CHANGED
@@ -3,4 +3,5 @@ gradio
3
  transformers
4
  numpy
5
  bitsandbytes
6
- newspaper3k
 
 
3
  transformers
4
  numpy
5
  bitsandbytes
6
+ newspaper3k
7
+ accelerate