Samurai719214 commited on
Commit
d101153
·
verified ·
1 Parent(s): d8cb255

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -17,7 +17,7 @@ def generate_full_story(excerpt: str) -> str:
17
  to generate the complete story that includes Parv, Key Event, Section and the story continuation.
18
  """
19
  # Tokenize the user-provided excerpt.
20
- encoded_input = tokenizer(excerpt, return_tensors="pt")
21
  # Move tensors to the appropriate device.
22
  encoded_input = {k: v.to(device) for k, v in encoded_input.items()}
23
 
@@ -34,7 +34,7 @@ def generate_full_story(excerpt: str) -> str:
34
  )
35
 
36
  # Decode the generated sequence.
37
- generated_text = tokenizer.decode(output.sequences[0], skip_special_tokens=True)
38
 
39
  return generated_text
40
 
@@ -46,9 +46,9 @@ interface = gr.Interface(
46
  label = "Incomplete Story Excerpt",
47
  placeholder = "Enter your incomplete story excerpt here..."
48
  ),
49
- outputs=gr.Textbox(label = "Complete Story with Details"),
50
- title="Mythology Storyteller",
51
- description=(
52
  "Enter an incomplete story excerpt. "
53
  "The model will generate a complete the complete chapter."
54
  )
 
17
  to generate the complete story that includes Parv, Key Event, Section and the story continuation.
18
  """
19
  # Tokenize the user-provided excerpt.
20
+ encoded_input = tokenizer(excerpt, return_tensors = "pt")
21
  # Move tensors to the appropriate device.
22
  encoded_input = {k: v.to(device) for k, v in encoded_input.items()}
23
 
 
34
  )
35
 
36
  # Decode the generated sequence.
37
+ generated_text = tokenizer.decode(output.sequences[0], skip_special_tokens = True)
38
 
39
  return generated_text
40
 
 
46
  label = "Incomplete Story Excerpt",
47
  placeholder = "Enter your incomplete story excerpt here..."
48
  ),
49
+ outputs = gr.Textbox(label = "Complete Story with Details"),
50
+ title = "🏺 Mythology Storyteller",
51
+ description = (
52
  "Enter an incomplete story excerpt. "
53
  "The model will generate a complete the complete chapter."
54
  )