Marcus Posey commited on
Commit
c5da24c
·
1 Parent(s): 84f9040

Temporary fix for model reinitialization

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -69,6 +69,12 @@ class ModelManager:
69
 
70
  def get_all_model_responses(self, prompt):
71
  responses = []
 
 
 
 
 
 
72
  for model in ['A', 'B', 'C']:
73
  response = self.get_model_response(f"Model_{model}", prompt)
74
  model_responses[f"Model_{model}"] = response
 
69
 
70
  def get_all_model_responses(self, prompt):
71
  responses = []
72
+ model_responses.clear()
73
+
74
+ self.model_A = Client("mep296/llama-3-8b-rephrase-quality")
75
+ self.model_B = Client("mep296/llama-3-8b-rephrase-summarize-quality")
76
+ self.model_C = Client("mep296/llama-3-8b-entigraph-quality")
77
+
78
  for model in ['A', 'B', 'C']:
79
  response = self.get_model_response(f"Model_{model}", prompt)
80
  model_responses[f"Model_{model}"] = response