krisha05 commited on
Commit
0d44900
·
verified ·
1 Parent(s): 05d09ee

created app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoTokenizer, AutoModelForCausalLM
2
+ import transformers
3
+ import torch
4
+
5
+ model = "tiiuae/falcon-7b"
6
+
7
+ tokenizer = AutoTokenizer.from_pretrained(model)
8
+ pipeline = transformers.pipeline(
9
+ "text-generation",
10
+ model=model,
11
+ tokenizer=tokenizer,
12
+ torch_dtype=torch.bfloat16,
13
+ trust_remote_code=True,
14
+ device_map="auto",
15
+ )
16
+ sequences = pipeline(
17
+ "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
18
+ max_length=200,
19
+ do_sample=True,
20
+ top_k=10,
21
+ num_return_sequences=1,
22
+ eos_token_id=tokenizer.eos_token_id,
23
+ )
24
+ for seq in sequences:
25
+ print(f"Result: {seq['generated_text']}")