Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
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-
|
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)
|