ranamhamoud commited on
Commit
e7fea68
·
verified ·
1 Parent(s): 92feef3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -69,9 +69,11 @@ def process_text(text):
69
 
70
  return text
71
 
 
72
  # Gradio Function
73
  @spaces.GPU
74
  def generate(
 
75
  message: str,
76
  chat_history: list[tuple[str, str]],
77
  max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
@@ -129,7 +131,7 @@ def generate(
129
  chat_interface = gr.ChatInterface(
130
  fn=generate,
131
  stop_btn=None,
132
- inputs=[model_selector = gr.Dropdown(model_choice=["Storytell", "HF Meta Llama 7b Chat"], label="Choose Model")]
133
  examples=[
134
  ["Can you explain briefly to me what is the Python programming language?"],
135
  ["Could you please provide an explanation about the concept of recursion?"],
 
69
 
70
  return text
71
 
72
+
73
  # Gradio Function
74
  @spaces.GPU
75
  def generate(
76
+ model_choice: str,
77
  message: str,
78
  chat_history: list[tuple[str, str]],
79
  max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
 
131
  chat_interface = gr.ChatInterface(
132
  fn=generate,
133
  stop_btn=None,
134
+ inputs=[model_choice = gr.Dropdown(["Storytell", "HF Meta Llama 7b Chat"], label="Choose Model")]
135
  examples=[
136
  ["Can you explain briefly to me what is the Python programming language?"],
137
  ["Could you please provide an explanation about the concept of recursion?"],