Update app.py

app.py

@@ -22,7 +22,7 @@ import pandas as pd
 
 import gradio as gr
 
-model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
+model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
 
 tokenizer = AutoTokenizer.from_pretrained(
     model_name,
@@ -44,8 +44,8 @@ quantization_config = BitsAndBytesConfig(
 
 # Get the model
 llm = HuggingFaceLLM(
-    model_name="meta-llama/Meta-Llama-3.1-8B-Instruct",
-    tokenizer_name="meta-llama/Meta-Llama-3.1-8B-Instruct",
+    model_name="meta-llama/Meta-Llama-3-8B-Instruct",
+    tokenizer_name="meta-llama/Meta-Llama-3-8B-Instruct",
     model_kwargs={
         "token": token_r,
         "quantization_config": quantization_config
@@ -248,9 +248,9 @@ demo = gr.Interface(
     inputs=prompt_box,
     outputs=output_box,
     # allow_flagging='auto',
-    # allow_flagging="manual",
-    # flagging_options=["appropriate","inappropriate","incorrect",],
-    # flagging_callback=hf_writer,
+    allow_flagging="manual",
+    flagging_options=["appropriate","inappropriate","incorrect",],
+    flagging_callback=hf_writer,
     # live=True
 )