abdo-Mansour commited on
Commit
6fcfebc
·
1 Parent(s): 251790a

eh suprise

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -196,7 +196,7 @@ def webpage_to_json(content: str, is_url: bool, schema: BaseModel) -> Dict[str,
196
  preprocessor = BasicPreprocessor(config={'keep_tags': True})
197
  try:
198
  # llm = GeminiLLMClient(config={'api_key': os.getenv('GEMINI_API_KEY')})
199
- llm = NvidiaLLMClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'qwen/qwen2.5-7b-instruct'})
200
  # reranker = NvidiaRerankerClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'nv-rerank-qa-mistral-4b:1'})\
201
  reranker = ModalRerankerClient("https://abdulrahmanmfam2003--qwen3-reranker-rerank.modal.run")
202
  except Exception as e:
 
196
  preprocessor = BasicPreprocessor(config={'keep_tags': True})
197
  try:
198
  # llm = GeminiLLMClient(config={'api_key': os.getenv('GEMINI_API_KEY')})
199
+ llm = NvidiaLLMClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'google/gemma-3n-e2b-it'})
200
  # reranker = NvidiaRerankerClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'nv-rerank-qa-mistral-4b:1'})\
201
  reranker = ModalRerankerClient("https://abdulrahmanmfam2003--qwen3-reranker-rerank.modal.run")
202
  except Exception as e: