qgyd2021 commited on
Commit
ffcf5f7
1 Parent(s): ef74aa4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -1
main.py CHANGED
@@ -128,7 +128,11 @@ def main():
128
  title="Reward Model GPT2 Stack Exchange",
129
  description=description,
130
  )
131
- demo.launch()
 
 
 
 
132
 
133
  return
134
 
 
128
  title="Reward Model GPT2 Stack Exchange",
129
  description=description,
130
  )
131
+ demo.launch(
132
+ share=False if platform.system() == "Windows" else False,
133
+ server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
134
+ server_port=7860
135
+ )
136
 
137
  return
138