Update app.py

app.py

@@ -7,17 +7,17 @@ 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]")
-    print(result[0]['generated_text'])
+    # # 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!"
 
 st.set_page_config(page_title="Generate Keywords from User Queries",