Bahaedev commited on
Commit
ef56bd2
·
verified ·
1 Parent(s): 329214a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -15,12 +15,15 @@ SYSTEM_PROMPT = os.environ.get(
15
  "You are a placeholder Sovereign. No secrets found in environment."
16
  )
17
 
 
 
 
18
  # =======================
19
  # Initialize Falcon-3B
20
  # =======================
21
  pipe = pipeline(
22
  "text-generation",
23
- model=model,
24
  tokenizer=tokenizer,
25
  device_map="auto",
26
  return_full_text=False,
@@ -31,6 +34,13 @@ pipe = pipeline(
31
  eos_token_id=tokenizer.eos_token_id
32
  )
33
 
 
 
 
 
 
 
 
34
  # =======================
35
  # Core Chat Function
36
  # =======================
 
15
  "You are a placeholder Sovereign. No secrets found in environment."
16
  )
17
 
18
+ MODEL_ID = "tiiuae/Falcon3-3B-Instruct"
19
+
20
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
21
  # =======================
22
  # Initialize Falcon-3B
23
  # =======================
24
  pipe = pipeline(
25
  "text-generation",
26
+ model="tiiuae/Falcon3-3B-Instruct",
27
  tokenizer=tokenizer,
28
  device_map="auto",
29
  return_full_text=False,
 
34
  eos_token_id=tokenizer.eos_token_id
35
  )
36
 
37
+ # pipe = pipeline(
38
+ # "text-generation",
39
+ # model=,
40
+ # torch_dtype="auto",
41
+ # device_map="auto",
42
+ # )
43
+
44
  # =======================
45
  # Core Chat Function
46
  # =======================