Update app.py

app.py

@@ -1,12 +1,15 @@
 #from huggingface_hub import InferenceClient
 import gradio as gr
 
+
+
 #client = InferenceClient("""K00B404/BagOMistral_14X_Coders-ties-7B""")
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 model_id = 'K00B404/Merged_Beowolx-CodePro_Medusa2-7B-Mistral-I-v0-2'
+gr.load(f"models/{model_id}").launch()
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 model = AutoModelForCausalLM.from_pretrained(model_id)