Web3Daily commited on
Commit
027f076
1 Parent(s): 28fd333

Update app.py

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