Update app.py

app.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import random
 import argparse
 import functools
 import os
@@ -68,22 +67,8 @@ def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
         model(z, label, force_fp32=True)
     return model
    
-def predict(text):
-  return pipe(text)[0]["translation_text"]
-
-def chat(message, history):
-    history = history or []
-    message = message.lower()
-    if message.startswith("how many"):
-        response = random.randint(1, 10)
-    elif message.startswith("how"):
-        response = random.choice(["Great", "Good", "Okay", "Bad"])
-    elif message.startswith("where"):
-        response = random.choice(["Here", "There", "Somewhere"])
-    else:
-        response = "I don't know"
-    history.append((message, response))
-    return history, history
+def greet(name):
+    return "Hello " + name + "!"
 
 
 def main():
@@ -128,13 +113,7 @@ def main():
         #server_port=args.port,
         #share=args.share,
     #)
-    chatbot = gr.Chatbot().style(color_map=("green", "pink"))
-    demo = gr.Interface(
-        chat,
-        ["text", "state"],
-        [chatbot, "state"],
-        allow_flagging="never",
-    )
+    demo = gr.Interface(fn=greet, inputs="text", outputs="text")
     demo.launch()
     
 if __name__ == '__main__':