Update app.py

app.py

@@ -3,7 +3,7 @@ import random
 import gradio as gr
 import sentencepiece as spm
 import numpy as np
-import tflite_runtime.interpreter as tflite
+import tensorflow as tf
 
 def custom_pad_sequences(sequences, maxlen, padding='pre', value=0):
     """
@@ -100,7 +100,7 @@ def generateTerrariaNames(amount, max_length=30, temperature=0.5, seed_text=""):
     max_seq_len = 12 # For skyrim = 13, for terraria = 12
 
     # Load TFLite model
-    interpreter = tflite.Interpreter(model_path="models/dungen_terraria_model.tflite")
+    interpreter = tf.lite.Interpreter(model_path="models/dungen_terraria_model.tflite")
     interpreter.allocate_tensors()
 
     # Use the function to generate a name
@@ -112,7 +112,7 @@ def generateTerrariaNames(amount, max_length=30, temperature=0.5, seed_text=""):
 
 demo = gr.Interface(
     fn=generateTerrariaNames,
-    inputs=[gr.Number(1,25), gr.Slider(10, 60), gr.Slider(0.01, 1), gr.Text(0,10)],
+    inputs=[gr.Number(1,25), gr.Slider(10, 60) gr.Slider(0.01, 1), gr.Text(0,10)],
     outputs=["text"],
 )