Update app.py

app.py

@@ -7,18 +7,18 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 def getLLamaresponse(input_text):
     ### LLama2 model
     # Load the fine-tuned model and tokenizer
-    # model_name = "Jithendra-k/interACT_LLM"
-    # model = AutoModelForCausalLM.from_pretrained(model_name)
-    # tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model_name = "Jithendra-k/interACT_LLM"
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-    # # Define the input prompt
-    # #prompt = "I want to drink water"
+    # Define the input prompt
+    #prompt = "I want to drink water"
 
-    # # Run text generation pipeline with the model
-    # pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=50, do_sample=True)
-    # result = pipe(f"<s>[INST] {input_text} [/INST]")
+    # Run text generation pipeline with the model
+    pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=50, do_sample=True)
+    result = pipe(f"<s>[INST] {input_text} [/INST]")
     # print(result[0]['generated_text'])
-    return "Hello World!"
+    return result[0]['generated_text']
 
 st.set_page_config(page_title="Generate Keywords from User Queries",
                    page_icon='🤖',