Hugging Face's logo

Spaces:
Stevross
/
Astrid-Assistant
Runtime error

App Files Community
Stevross commited on
Commit
9ea43cc
·
1 Parent(s): 9b40584

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  # Initialize the model and pipeline
5
  generate_text = pipeline(
6
- model="Stevross/Astrid-3B,
7
  task="text-generation",
8
  trust_remote_code=True,
9
  device=0 # Use GPU if available, otherwise CPU
 
3
 
4
  # Initialize the model and pipeline
5
  generate_text = pipeline(
6
+ model="Stevross/Astrid-3B",
7
  task="text-generation",
8
  trust_remote_code=True,
9
  device=0 # Use GPU if available, otherwise CPU