acecalisto3 commited on
Commit
33dbea2
·
verified ·
1 Parent(s): 702dc0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -13
app.py CHANGED
@@ -70,18 +70,20 @@ def fetch_github_issues(github_api_token: str, github_username: str, github_repo
70
  return issues
71
 
72
  def respond(
73
- command: str,
74
- history: list[tuple[str, str]],
75
- system_message: str,
76
- max_tokens: int,
77
- temperature: float,
78
- top_p: float,
79
- github_api_token: str,
80
- github_username: str,
81
- github_repository: str,
82
- selected_model: str,
83
- severity: str,
84
- programming_language: str,
 
 
85
  ) -> str:
86
  # Initialize the model
87
  model = pipeline("text-generation", model=selected_model)
@@ -99,7 +101,6 @@ def respond(
99
  assistant_response = response[0]['generated_text'].strip()
100
 
101
  return assistant_response
102
-
103
  class MyChatbot(gr.ChatInterface):
104
  def __init__(self, fn, *args, **kwargs):
105
  super().__init__(fn, *args, **kwargs)
 
70
  return issues
71
 
72
  def respond(
73
+ command,
74
+ history,
75
+ system_message,
76
+ max_tokens,
77
+ temperature,
78
+ top_p,
79
+ github_api_token,
80
+ github_username,
81
+ github_repository,
82
+ selected_model,
83
+ severity,
84
+ programming_language,
85
+ *args,
86
+ **kwargs,
87
  ) -> str:
88
  # Initialize the model
89
  model = pipeline("text-generation", model=selected_model)
 
101
  assistant_response = response[0]['generated_text'].strip()
102
 
103
  return assistant_response
 
104
  class MyChatbot(gr.ChatInterface):
105
  def __init__(self, fn, *args, **kwargs):
106
  super().__init__(fn, *args, **kwargs)