Update app.py

app.py

@@ -95,7 +95,7 @@ response = requests.get(url)
 with open(local_filename, 'wb') as f:
     f.write(response.content)
 '''
-model = tf.keras.layers.TFSMLayer('my_model', call_endpoint='serving_default')
+model = tf.keras.models.load('my_model')
 
 
 
@@ -116,7 +116,7 @@ def generate_arabic_text(seed_text, next_words=50):
     for _ in range(next_words):
         token_list = tokenizer.encode(generated_text, add_special_tokens=False)
         token_list = pad_sequences([token_list], maxlen=max_sequence_len-1, padding='pre')
-        predicted = np.argmax(model(token_list), axis=-1)
+        predicted = np.argmax(model.predict(token_list), axis=-1)
         output_word = tokenizer.decode(predicted[0])
         generated_text += " " + output_word
     reconnected_text = generated_text.replace(" ##", "")