Avamalton commited on
Commit
550ba8f
Β·
1 Parent(s): 23881bb
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Load model dari Hugging Face tanpa device_map untuk menghindari error di Spaces
5
  generator = pipeline(
6
  "text-generation",
7
- model="tiiuae/falcon-rw-1b",
8
  trust_remote_code=True # boleh dihapus jika pakai transformers terbaru
9
  )
10
 
 
4
  # Load model dari Hugging Face tanpa device_map untuk menghindari error di Spaces
5
  generator = pipeline(
6
  "text-generation",
7
+ model="tiiuae/Falcon-H1-1.5B-Instruct",
8
  trust_remote_code=True # boleh dihapus jika pakai transformers terbaru
9
  )
10