abdo-Mansour commited on
Commit
7c08c22
·
verified ·
1 Parent(s): d5f9622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -224,7 +224,7 @@ def webpage_to_json(content: str, is_url: bool, schema: BaseModel) -> Dict[str,
224
  preprocessor = BasicPreprocessor(config={'keep_tags': True})
225
  try:
226
  # llm = GeminiLLMClient(config={'api_key': os.getenv('GEMINI_API_KEY')})
227
- llm = NvidiaLLMClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'google/gemma-3n-e4b-it'})
228
  except Exception as e:
229
  return {"error": f"Failed to initialize LLM client: {str(e)}"}
230
 
 
224
  preprocessor = BasicPreprocessor(config={'keep_tags': True})
225
  try:
226
  # llm = GeminiLLMClient(config={'api_key': os.getenv('GEMINI_API_KEY')})
227
+ llm = NvidiaLLMClient(config={'api_key': os.getenv('NVIDIA_API_KEY'),'model_name': 'google/gemma-3n-e2b-it'})
228
  except Exception as e:
229
  return {"error": f"Failed to initialize LLM client: {str(e)}"}
230