Update app.py

app.py

@@ -9,12 +9,13 @@ if torch.cuda.is_available():
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
 @spaces.GPU
-def predict(text):
-    return pipe(text)
+def predict(prompt, temperature, max_length):
+    return pipe(prompt, temperature=temperature, max_length=max_length)[0]["generated_text"]
 
 demo = gr.Interface(
     fn=predict,
-    inputs=["text"],
+    title="mGPT-13B Demo",
+    inputs=["text", gr.Slider(minimum=0.01, maximum=1.0, value=0.7), gr.Slider(minimum=1, maximum=1024, value=50)],
     outputs=["text"],
 )