Web3Daily commited on
Commit
0870e71
1 Parent(s): f1dc173

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  def chat_with_gpt(input_text):
5
  model_name = "text-davinci-003" # Replace this with the appropriate GPT model you want to use.
6
  chat_gpt = pipeline("text-generation", model=model_name, device=0)
7
- generated_text = chat_gpt(input_text, max_length=150, do_sample=True, top_k=50, top_p=0.95)
8
  return generated_text[0]["generated_text"]
9
 
10
  # Gradio user interface
 
4
  def chat_with_gpt(input_text):
5
  model_name = "text-davinci-003" # Replace this with the appropriate GPT model you want to use.
6
  chat_gpt = pipeline("text-generation", model=model_name, device=0)
7
+ generated_text = chat_gpt(input_text, max_length=1000, do_sample=True, top_k=500, top_p=1)
8
  return generated_text[0]["generated_text"]
9
 
10
  # Gradio user interface