Hugging Face's logo

Spaces:
aixsatoshi
/
sarashina2.1-1b-sft-CPU
Sleeping

App Files Community
aixsatoshi commited on
Commit
d139d80
•
1 Parent(s): a352461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,17 +6,17 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
6
  import gradio as gr
7
  from threading import Thread
8
 
9
- model_id = "ibm-granite/granite-3.0-1b-a400m-instruct"
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_id,
13
  device_map="cpu",
14
  )
15
 
16
- TITLE = "<h1><center>ibm-granite/granite-3.0-1b-a400m-instruct by CPU</center></h1>"
17
 
18
  DESCRIPTION = """
19
- <h3>MODEL: <a href="https://huggingface.co/ibm-granite/granite-3.0-1b-a400m-instruct">ibm-granite/granite-3.0-1b-a400m-instruct</a></h3>
20
  <center>
21
  <p>This model is designed for conversational interactions.</p>
22
  </center>
 
6
  import gradio as gr
7
  from threading import Thread
8
 
9
+ model_id = "Aratako/sarashina2.1-1b-sft"
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(
12
  model_id,
13
  device_map="cpu",
14
  )
15
 
16
+ TITLE = "<h1><center>Aratako/sarashina2.1-1b-sft by CPU</center></h1>"
17
 
18
  DESCRIPTION = """
19
+ <h3>MODEL: <a href="https://huggingface.co/Aratako/sarashina2.1-1b-sft">Aratako/sarashina2.1-1b-sft</a></h3>
20
  <center>
21
  <p>This model is designed for conversational interactions.</p>
22
  </center>