SanjayRohith commited on
Commit
3e852bc
1 Parent(s): 854be63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
  title = "Text Generator"
6
  description = "This text generator has been trained to chat and to respond to natural language instructions."
7
 
8
- ans = pipeline(model="gpt2", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
9
 
10
  def answer(query):
11
  out = ans(query)
 
5
  title = "Text Generator"
6
  description = "This text generator has been trained to chat and to respond to natural language instructions."
7
 
8
+ ans = pipeline(model="databricks/dolly-v2-3", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
9
 
10
  def answer(query):
11
  out = ans(query)