Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
+import tensorflow as tf
+import numpy as np
 
 from transformers import AutoModelForSequenceClassification
 
 model = AutoModelForSequenceClassification.from_pretrained("Alesteba/deep_model_07")
 
-def predict(prompt):
-    completion = model(prompt)[0]["generated_text"]
-    return completion
+from huggingface_hub import from_pretrained_keras
+model = from_pretrained_keras("Alesteba/deep_model_07")
 
+def predict(prompt):
+    return model.predict('prompt')[0]
+    
 gr.Interface(fn=predict, inputs="text", outputs="text").launch()