GIGGeorg commited on
Commit
1f26b4f
·
verified ·
1 Parent(s): 6fc7e9b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=100)
2
+ prompt = "My name is Harry and I live in Winnipeg. My phone number is ummm 204 no 203, ahh 4344, no 4355"
3
+ result = pipe(f"<s>[INST] {prompt} [/INST]")
4
+
5
+ # Print the generated text
6
+ print(result[0]['generated_text'])