Spaces:
Runtime error
Runtime error
Marcus Posey
commited on
Commit
·
c5da24c
1
Parent(s):
84f9040
Temporary fix for model reinitialization
Browse files
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
|