Andyrasika commited on
Commit
01f2f78
1 Parent(s): 327d5b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,13 +19,13 @@ def create_spark_dataframe(text):
19
  return df
20
 
21
  # Read the "War and Peace" text file and create Spark DataFrame
22
- with open('/content/data/war_and_peace.txt', 'r') as file:
23
  text = file.read()
24
  df_chapters = create_spark_dataframe(text)
25
 
26
  # Define the Llama models
27
- MODEL_Q8_0 = Llama(model_path="./models/llama-2-7b-chat.ggmlv3.q8_0.bin", n_ctx=8192, n_batch=512)
28
- MODEL_Q2_K = Llama(model_path="./models/llama-2-7b-chat.ggmlv3.q2_K.bin", n_ctx=8192, n_batch=512)
29
 
30
  # Function to summarize a chapter using the selected model
31
  def llama2_summarize(chapter_text, model_version):
 
19
  return df
20
 
21
  # Read the "War and Peace" text file and create Spark DataFrame
22
+ with open('war_and_peace.txt', 'r') as file:
23
  text = file.read()
24
  df_chapters = create_spark_dataframe(text)
25
 
26
  # Define the Llama models
27
+ MODEL_Q8_0 = Llama(model_path="llama-2-7b-chat.ggmlv3.q8_0.bin", n_ctx=8192, n_batch=512)
28
+ MODEL_Q2_K = Llama(model_path="llama-2-7b-chat.ggmlv3.q2_K.bin", n_ctx=8192, n_batch=512)
29
 
30
  # Function to summarize a chapter using the selected model
31
  def llama2_summarize(chapter_text, model_version):