AdamyaG commited on
Commit
1cafb1d
1 Parent(s): 2d6da1e

Update chatbot.py

Browse files
Files changed (1) hide show
  1. chatbot.py +6 -6
chatbot.py CHANGED
@@ -164,13 +164,13 @@ def search(query):
164
  return all_results
165
 
166
 
167
- def image_gen(prompt):
168
- client = Client("KingNish/Image-Gen-Pro")
169
- return client.predict("Image Generation",None, prompt, api_name="/image_gen_pro")
170
 
171
- def video_gen(prompt):
172
- client = Client("KingNish/Instant-Video")
173
- return client.predict(prompt, api_name="/instant_video")
174
 
175
  def llava(user_prompt, chat_history):
176
  if user_prompt["files"]:
 
164
  return all_results
165
 
166
 
167
+ # def image_gen(prompt):
168
+ # client = Client("KingNish/Image-Gen-Pro")
169
+ # return client.predict("Image Generation",None, prompt, api_name="/image_gen_pro")
170
 
171
+ # def video_gen(prompt):
172
+ # client = Client("KingNish/Instant-Video")
173
+ # return client.predict(prompt, api_name="/instant_video")
174
 
175
  def llava(user_prompt, chat_history):
176
  if user_prompt["files"]: