tranquilkd commited on
Commit
9b3b0a3
·
1 Parent(s): f1a8924

corrected app

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import torch
3
  import gradio as gr
4
  from typing import Optional
@@ -73,6 +74,13 @@ def generate_text(prompt,
73
  top_p: float = 0.7
74
  ):
75
  """Generate text from a prompt"""
 
 
 
 
 
 
 
76
  input_ids = tokenizer(prompt,
77
  padding=True,
78
  truncation=True,
@@ -92,12 +100,9 @@ def generate_text(prompt,
92
 
93
  return tokenizer.decode(generated[0], skip_special_tokens=True)
94
 
95
- # Gradio interface
96
- def gradio_interface(prompt, max_length, temperature, top_k):
97
- return generate_text(prompt, int(max_length), float(temperature), int(top_k))
98
 
99
  iface = gr.Interface(
100
- fn=gradio_interface,
101
  inputs=[
102
  gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
103
  gr.Slider(minimum=10, maximum=500, value=28, label="Min Length"),
 
1
  import os
2
+ import tempfile
3
  import torch
4
  import gradio as gr
5
  from typing import Optional
 
74
  top_p: float = 0.7
75
  ):
76
  """Generate text from a prompt"""
77
+
78
+ min_length = int(max_length)
79
+ max_length = int(max_length)
80
+ temperature = float(temperature)
81
+ top_k = int(top_k)
82
+ top_p = float(top_p)
83
+
84
  input_ids = tokenizer(prompt,
85
  padding=True,
86
  truncation=True,
 
100
 
101
  return tokenizer.decode(generated[0], skip_special_tokens=True)
102
 
 
 
 
103
 
104
  iface = gr.Interface(
105
+ fn=generate_text,
106
  inputs=[
107
  gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
108
  gr.Slider(minimum=10, maximum=500, value=28, label="Min Length"),