srijaydeshpande commited on
Commit
9ce69b5
·
verified ·
1 Parent(s): 3f31391

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -158,8 +158,8 @@ def pdf_to_text(cv_file, job_description, llm_type='Fine tuned Llama3', maxtoken
158
  result = converter.convert(cv_file)
159
  cv_text = result.document.export_to_markdown()
160
 
161
- if(llm_type=='Fine tuned Llama3'):
162
- llm = Llama(
163
  model_path="models/" + model_id,
164
  flash_attn=True,
165
  n_gpu_layers=81,
 
158
  result = converter.convert(cv_file)
159
  cv_text = result.document.export_to_markdown()
160
 
161
+ # if(llm_type=='Fine tuned Llama3'):
162
+ llm = Llama(
163
  model_path="models/" + model_id,
164
  flash_attn=True,
165
  n_gpu_layers=81,