arjunanand13 commited on
Commit
6dfb8a2
·
verified ·
1 Parent(s): 300f75a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -83,8 +83,8 @@ class DocumentRetrievalAndGeneration:
83
  device_map="auto",
84
  quantization_config=quantization_config,
85
  token=hf_token
86
- ) quantization_config=quantization_config
87
-
88
  return tokenizer, model
89
 
90
  def generate_response_with_timeout(self, input_ids, max_new_tokens=1000):
 
83
  device_map="auto",
84
  quantization_config=quantization_config,
85
  token=hf_token
86
+ quantization_config=quantization_config
87
+ )
88
  return tokenizer, model
89
 
90
  def generate_response_with_timeout(self, input_ids, max_new_tokens=1000):