legolasyiu commited on
Commit
914b14a
·
verified ·
1 Parent(s): 532ea28

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +0 -11
src/streamlit_app.py CHANGED
@@ -1,5 +1,4 @@
1
  import os
2
- import torch
3
  import streamlit as st
4
  from typing import Literal
5
  from tavily import TavilyClient
@@ -85,16 +84,6 @@ def load_model(provider: str):
85
  os.environ["GOOGLE_API_KEY"] = gemini_api_key
86
  return init_chat_model(model="gemini-2.5-pro")
87
 
88
- elif provider == "Hugging Face":
89
- device = 0 if torch.cuda.is_available() else -1
90
- llm = HuggingFacePipeline.from_model_id(
91
- model_id="EpistemeAI/Episteme-gptoss-20b-RL",
92
- task="text-generation",
93
- device=device,
94
- pipeline_kwargs={"max_new_tokens": 8000},
95
- )
96
- return ChatHuggingFace(llm=llm)
97
-
98
  else:
99
  st.error("Unknown provider selected.")
100
  st.stop()
 
1
  import os
 
2
  import streamlit as st
3
  from typing import Literal
4
  from tavily import TavilyClient
 
84
  os.environ["GOOGLE_API_KEY"] = gemini_api_key
85
  return init_chat_model(model="gemini-2.5-pro")
86
 
 
 
 
 
 
 
 
 
 
 
87
  else:
88
  st.error("Unknown provider selected.")
89
  st.stop()