dj86 commited on
Commit
0e2c65f
1 Parent(s): 12514b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import os
2
  import json
3
  import gradio as gr
4
- import openai
5
  import requests
6
  import csv
7
  import argparse
@@ -11,7 +10,6 @@ from models.vlog4debate import Debate
11
  from utils.utils import download_video
12
 
13
  #prompt_templates = {"Default ChatGPT": ""}
14
- os.environ['HF_HOME'] = '/root/autodl-tmp/cache/'
15
 
16
  parser = argparse.ArgumentParser()
17
  parser.add_argument('--video_path', default='examples/training.mp4')
@@ -81,7 +79,7 @@ def submit_message_debate(prompt, state):
81
  while debate_topic == "":
82
  debate_topic = prompt
83
 
84
- config = json.load(open("/root/autodl-tmp/cache/config4all.json", "r"))
85
  config['debate_topic'] = debate_topic
86
 
87
  debate = Debate(num_players=3, config=config, temperature=0, sleep_time=0)
 
1
  import os
2
  import json
3
  import gradio as gr
 
4
  import requests
5
  import csv
6
  import argparse
 
10
  from utils.utils import download_video
11
 
12
  #prompt_templates = {"Default ChatGPT": ""}
 
13
 
14
  parser = argparse.ArgumentParser()
15
  parser.add_argument('--video_path', default='examples/training.mp4')
 
79
  while debate_topic == "":
80
  debate_topic = prompt
81
 
82
+ config = json.load(open("./config4all.json", "r"))
83
  config['debate_topic'] = debate_topic
84
 
85
  debate = Debate(num_players=3, config=config, temperature=0, sleep_time=0)