mohamedashraf11 commited on
Commit
3de2665
·
verified ·
1 Parent(s): 2cacb26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ class HadithChatApp:
29
  )
30
 
31
  # Initialize the language model (AceGPT-v2-32B-Chat)
32
- self.tokenizer = AutoTokenizer.from_pretrained("FreedomIntelligence/AceGPT-v2-32B-Chat")
33
- self.model = AutoModelForCausalLM.from_pretrained("FreedomIntelligence/AceGPT-v2-32B-Chat")
34
 
35
  def extract_zip(self, zip_path, extract_to):
36
  """Extract a ZIP file to a specified directory."""
 
29
  )
30
 
31
  # Initialize the language model (AceGPT-v2-32B-Chat)
32
+ self.tokenizer = AutoTokenizer.from_pretrained("FreedomIntelligence/AceGPT-7B")
33
+ self.model = AutoModelForCausalLM.from_pretrained("FreedomIntelligence/AceGPT-7B")
34
 
35
  def extract_zip(self, zip_path, extract_to):
36
  """Extract a ZIP file to a specified directory."""