nirmalaag commited on
Commit
34d4635
1 Parent(s): 4b12318

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,8 +61,8 @@ def main():
61
  query = st.text_input("Ask questions about your PDF file:")
62
 
63
  if query:
64
- torch.cuda.empty_cache()
65
- PATH = 'model/'
66
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
67
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
68
  llm = huggingface_hub.HuggingFaceHub(repo_id="CohereForAI/aya-101",
 
61
  query = st.text_input("Ask questions about your PDF file:")
62
 
63
  if query:
64
+
65
+ #PATH = 'model/'
66
  #llm = AutoModelForCausalLM.from_pretrained("CohereForAI/aya-101")
67
  # llm = AutoModelForCausalLM.from_pretrained(PATH,local_files_only=True)
68
  llm = huggingface_hub.HuggingFaceHub(repo_id="CohereForAI/aya-101",