Update app.py

app.py

@@ -57,7 +57,7 @@ def generate_output(input_text,model):
     output = ""
     # Initialize the FastLanguageModel
     model, tokenizer = FastLanguageModel.from_pretrained(
-        model_name = model,
+        model_name = "DataIntelligenceTeam/NER-Phi-3-mini-4k-instruct",
         max_seq_length = 2500,
         dtype = None,
         load_in_4bit = True,
@@ -113,7 +113,7 @@ model_options = ["DataIntelligenceTeam/NER-Phi-3-mini-4k-instruct"]
 iface = gr.Interface(fn=generate_output, 
                      inputs=[
                             gr.inputs.Textbox(label="Input Text"),
-                            gr.inputs.Dropdown(label="Select the Fine-tuned Model", choices=["DataIntelligenceTeam/NER-Phi-3-mini-4k-instruct","DataIntelligenceTeam/llama-3-8b-Instruct-bnb-4bit"])  
+                            #gr.inputs.Dropdown(label="Select the Fine-tuned Model", choices=["DataIntelligenceTeam/NER-Phi-3-mini-4k-instruct","DataIntelligenceTeam/llama-3-8b-Instruct-bnb-4bit"])  
                              ],
                      outputs=[
                             gr.outputs.Textbox(label="Original Output Text"),