sr5434 commited on
Commit
42965da
1 Parent(s): 9de6762

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def generate_text(prompt1, prompt2):
7
  model = "EleutherAI/gpt-neo-125M"
8
  summarization = pipeline("text-generation", model=model)
9
  # Concatenate the two prompts
10
- prompt = "Explain " + prompt1 + " to a " + prompt + " grader below."
11
  # Generate the text
12
  result = summarization(prompt, max_length=32)
13
  return result[0]["generated_text"]
 
7
  model = "EleutherAI/gpt-neo-125M"
8
  summarization = pipeline("text-generation", model=model)
9
  # Concatenate the two prompts
10
+ prompt = "Explain " + prompt1 + " to a " + prompt2 + " grader below."
11
  # Generate the text
12
  result = summarization(prompt, max_length=32)
13
  return result[0]["generated_text"]