Pooja P commited on
Commit
be9b79b
·
1 Parent(s): 6221859

used transformer instead of openai

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import pipeline
10
  import gradio as gr
11
 
12
  # Load the model
13
- generator = pipeline("text-generation", model="gpt-3.5-turbo")
14
 
15
  # Define your function
16
  def generate_text(prompt):
 
10
  import gradio as gr
11
 
12
  # Load the model
13
+ generator = pipeline("text-generation", model="Distilgpt2")
14
 
15
  # Define your function
16
  def generate_text(prompt):