Stevross commited on
Commit
9b40584
Β·
1 Parent(s): d507392

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-7B-Assistant-CPU",
7
  task="text-generation",
8
  trust_remote_code=True,
9
  device=0 # Use GPU if available, otherwise CPU
@@ -20,7 +20,7 @@ interface = gr.Interface(
20
  outputs=gr.outputs.Textbox(),
21
  live=True,
22
  title="Astrid-7B-Assistant-CPU Demo",
23
- description="An interactive demo of the Astrid-7B-Assistant-CPU model from Hugging Face."
24
  )
25
 
26
  interface.launch(share=True)
 
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
 
20
  outputs=gr.outputs.Textbox(),
21
  live=True,
22
  title="Astrid-7B-Assistant-CPU Demo",
23
+ description="An interactive demo of the Stevross/Astrid-3B model from Hugging Face."
24
  )
25
 
26
  interface.launch(share=True)