rish13 commited on
Commit
8849b7f
·
verified ·
1 Parent(s): d037267

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = pipeline("text-generation", model="rish13/polymers")
7
  def generate_response(prompt):
8
  # Generate text from the model
9
  response = model(prompt, max_length=150, num_return_sequences=1)
10
- return response[0]['generated_text']
11
 
12
  # Define the Gradio interface
13
  interface = gr.Interface(
 
7
  def generate_response(prompt):
8
  # Generate text from the model
9
  response = model(prompt, max_length=150, num_return_sequences=1)
10
+ return response[0]['generated_text']
11
 
12
  # Define the Gradio interface
13
  interface = gr.Interface(