Update app.py

app.py

@@ -12,7 +12,6 @@ description = "Gradio Demo for a miniature with GPT. To use it, simply add your
 tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
 
 model = from_pretrained_keras("keras-io/text-generation-miniature-gpt")
-  
 
 def tokenize_data(text):
     # Tokenize the review body
@@ -27,12 +26,7 @@ def tokenize_data(text):
 
 def generate_answers(text):
     inputs = tokenize_data(text)
-    results= model.generate(input_ids= inputs['input_ids'], attention_mask=inputs['attention_mask'], do_sample=True,
-                            max_length=120,
-                            top_k=120,
-                            top_p=0.98,
-                            early_stopping=True,
-                            num_return_sequences=1)
+    results= model.predict(inputs['input_ids'])
     answer = tokenizer.decode(results[0], skip_special_tokens=True)
     return answer