davidna22 commited on
Commit
916d30d
·
1 Parent(s): c8b4a92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -37,6 +37,13 @@ def init_app(api_key,
37
  diffusion_model,
38
  keyword_model,
39
  dtype):
 
 
 
 
 
 
 
40
  console_logs = StringIO()
41
  sys.stdout = console_logs
42
  model_class = model_class.replace(" ","").lower()
@@ -84,6 +91,9 @@ def run_simulation(topic,
84
  image_tokens=[],
85
  news_company="CNN",
86
  text_model_name="gpt-3.5-turbo-0301"):
 
 
 
87
  console_logs = StringIO()
88
  sys.stdout = console_logs
89
  filename = topic.replace(" ", "-") + ".csv"
 
37
  diffusion_model,
38
  keyword_model,
39
  dtype):
40
+
41
+ try:
42
+ openai.api_key = api_key
43
+ openai.Model.list()
44
+ except Exception as e:
45
+ gr.Error(str(e))
46
+
47
  console_logs = StringIO()
48
  sys.stdout = console_logs
49
  model_class = model_class.replace(" ","").lower()
 
91
  image_tokens=[],
92
  news_company="CNN",
93
  text_model_name="gpt-3.5-turbo-0301"):
94
+ if topic == "":
95
+ gr.Error("Topic must not be empty")
96
+
97
  console_logs = StringIO()
98
  sys.stdout = console_logs
99
  filename = topic.replace(" ", "-") + ".csv"