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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import gradio as gr
2
  from transformers import pipeline
3
 
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
 
2
  from transformers import pipeline
3
 
4
  def chat_with_gpt(input_text):
5
+ model_name = "openai-gpt" # 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