Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 
 import gradio as gr
 
-from text_generation import Client, InferenceAPIClient
+gr.Interface.load("models/OpenAssistant/oasst-sft-1-pythia-12b").launch()
 
 openchat_preprompt = (
     "\n<human>: Hi!\n<bot>: My name is multivac, model version is 0.15, part of an open-source kit for "
@@ -12,13 +12,6 @@ openchat_preprompt = (
 )
 
 
-def get_client(model: str):
-    if model == "Rallio67/joi2_20Be_instruct_alpha":
-        return Client(os.getenv("JOI_API_URL"))
-    if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
-        return Client(os.getenv("OPENCHAT_API_URL"))
-    return InferenceAPIClient(model, token=os.getenv("HF_TOKEN", None))
-
 
 def get_usernames(model: str):
     """
@@ -27,11 +20,7 @@ def get_usernames(model: str):
     """
     if model == "OpenAssistant/oasst-sft-1-pythia-12b":
         return "", "<|prompter|>", "<|assistant|>", "<|endoftext|>"
-    if model == "Rallio67/joi2_20Be_instruct_alpha":
-        return "", "User: ", "Joi: ", "\n\n"
-    if model == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
-        return openchat_preprompt, "<human>: ", "<bot>: ", "\n"
-    return "", "User: ", "Assistant: ", "\n"
+
 
 
 def predict(
@@ -161,7 +150,7 @@ def radio_on_change(
     )
 
 
-title = """<h1 align="center">🔥Large Language Model API 🚀Streaming🚀</h1>"""
+title = """<h1 align="center">🔥SAPIENS-IA🚀</h1>"""
 description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
 ```
 User: <utterance>