RamAnanth1 commited on
Commit
7947ceb
1 Parent(s): 3796a5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -3,10 +3,16 @@ import gradio as gr
3
  import os, json
4
  from loguru import logger
5
  import random
 
 
6
 
7
  session_token = os.environ.get('SessionToken')
8
  # logger.info(f"session_token_: {session_token}")
9
 
 
 
 
 
10
  def get_response_from_chatbot(text):
11
  try:
12
  api = ChatGPT(session_token)
@@ -130,7 +136,7 @@ with gr.Blocks(title='Talk to chatGPT') as demo:
130
  chatbot = gr.Chatbot(elem_id="chat_bot", visible=False).style(color_map=("green", "blue"))
131
  chatbot1 = gr.Chatbot(elem_id="chat_bot1").style(color_map=("green", "blue"))
132
  with gr.Row(elem_id="prompt_row"):
133
- prompt_input = gr.Textbox(lines=2, label="prompt",show_label=False)
134
  chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
135
  submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
136
  margin=True,
 
3
  import os, json
4
  from loguru import logger
5
  import random
6
+ import whisper
7
+ import torch
8
 
9
  session_token = os.environ.get('SessionToken')
10
  # logger.info(f"session_token_: {session_token}")
11
 
12
+ whisper_model = whisper.load_model("medium")
13
+
14
+ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
15
+
16
  def get_response_from_chatbot(text):
17
  try:
18
  api = ChatGPT(session_token)
 
136
  chatbot = gr.Chatbot(elem_id="chat_bot", visible=False).style(color_map=("green", "blue"))
137
  chatbot1 = gr.Chatbot(elem_id="chat_bot1").style(color_map=("green", "blue"))
138
  with gr.Row(elem_id="prompt_row"):
139
+ prompt_input = gr.Textbox(lines=2, label="Input text",show_label=True)
140
  chat_history = gr.Textbox(lines=4, label="prompt", visible=False)
141
  submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
142
  margin=True,