Spaces:
Sleeping
Sleeping
Commit
·
9b3b0a3
1
Parent(s):
f1a8924
corrected app
Browse files
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=
|
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"),
|