camparchimedes commited on
Commit
8c188ac
1 Parent(s): 43cb2b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -25,9 +25,9 @@ def respond(
25
  if uploaded_file is not None:
26
  with open(uploaded_file.name, "r") as f:
27
  file_content = f.read()
28
- messages.append({"role": "user", "content": file_content})
29
-
30
- messages.append({"role": "user", "content": message})
31
 
32
  response = ""
33
 
@@ -43,11 +43,11 @@ def respond(
43
  response += token
44
  yield response
45
 
46
- # Process
47
  if uploaded_file is not None:
48
  print(f"Uploaded file: {uploaded_file.name}")
49
 
50
- # CSV file
51
  if uploaded_file.name.endswith(".csv"):
52
  try:
53
  df = pd.read_csv(uploaded_file.name)
@@ -59,7 +59,6 @@ def respond(
59
  except Exception as e:
60
  print(f"Error loading CSV file: {e}")
61
 
62
- # text file
63
  elif uploaded_file.name.endswith(".txt"):
64
  try:
65
  with open(uploaded_file.name, "r") as f:
@@ -72,11 +71,13 @@ def respond(
72
  except Exception as e:
73
  print(f"Error loading text file: {e}")
74
 
 
75
  demo = gr.ChatInterface(
76
  respond,
77
  title="Nixie Steamcore, a hotbot!",
78
  additional_inputs=[
79
- gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
 
80
  gr.Slider(minimum=0.1, maximum=4.0, value=1.2, step=0.1, label="Temperature"),
81
  gr.Slider(
82
  minimum=0.1,
@@ -92,6 +93,7 @@ demo = gr.ChatInterface(
92
  if __name__ == "__main__":
93
  demo.launch()
94
 
 
95
  """
96
  import gradio as gr
97
  from huggingface_hub import InferenceClient
 
25
  if uploaded_file is not None:
26
  with open(uploaded_file.name, "r") as f:
27
  file_content = f.read()
28
+ messages.append({"role": "user", "content": f"{message}\n\nFile content:\n{file_content}"})
29
+ else:
30
+ messages.append({"role": "user", "content": message})
31
 
32
  response = ""
33
 
 
43
  response += token
44
  yield response
45
 
46
+
47
  if uploaded_file is not None:
48
  print(f"Uploaded file: {uploaded_file.name}")
49
 
50
+
51
  if uploaded_file.name.endswith(".csv"):
52
  try:
53
  df = pd.read_csv(uploaded_file.name)
 
59
  except Exception as e:
60
  print(f"Error loading CSV file: {e}")
61
 
 
62
  elif uploaded_file.name.endswith(".txt"):
63
  try:
64
  with open(uploaded_file.name, "r") as f:
 
71
  except Exception as e:
72
  print(f"Error loading text file: {e}")
73
 
74
+
75
  demo = gr.ChatInterface(
76
  respond,
77
  title="Nixie Steamcore, a hotbot!",
78
  additional_inputs=[
79
+ gr.Textbox(value="Nixie Steamcore, a hotbot!", label="System message"),
80
+ gr.Slider(minimum=1, maximum=2048, value=2048, step=1, label="Max new tokens"),
81
  gr.Slider(minimum=0.1, maximum=4.0, value=1.2, step=0.1, label="Temperature"),
82
  gr.Slider(
83
  minimum=0.1,
 
93
  if __name__ == "__main__":
94
  demo.launch()
95
 
96
+
97
  """
98
  import gradio as gr
99
  from huggingface_hub import InferenceClient