jat

app.py

@@ -5,14 +5,14 @@ from transformers import pipeline
 
 # Get the list of models from the Hugging Face Hub
 api = HfApi()
-models = api.list_models(author="gia-project", filter=ModelFilter(tags="text-generation"))
+models = api.list_models(author="jat-project", filter=ModelFilter(tags="text-generation"))
 models_names = [model.modelId for model in models]
 
 # Dictionary to store loaded models and their pipelines
 model_pipelines = {}
 
 # Load a default model initially
-default_model_name = "gia-project/gia2-small-untrained"
+default_model_name = "jat-project/jat-small"
 
 
 def generate_text(model_name, input_text):
@@ -42,11 +42,11 @@ def generate_text(model_name, input_text):
 iface = gr.Interface(
     fn=generate_text,  # Function to be called on user input
     inputs=[
-        Dropdown(models_names, label="Select Model", value="gia-project/gia2-small-untrained"),  # Select model
+        Dropdown(models_names, label="Select Model", value=default_model_name),  # Select model
         Textbox(lines=5, label="Input Text"),  # Textbox for entering text
     ],
     outputs=Textbox(label="Generated Text"),  # Textbox to display the generated text
-    title="GIA Text Generation",  # Title of the interface
+    title="JAT Text Generation",  # Title of the interface
 )
 
 # Launch the Gradio interface