Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -194,7 +194,7 @@ print("done")
|
|
194 |
#Testen des fine-tuned Modells
|
195 |
print("Predict")
|
196 |
login(token=os.environ["HF_ACCESS_TOKEN"]) #read access token!!!!!!!!!!
|
197 |
-
model_name = "test-tis-1"
|
198 |
tokenizer_neu = load_tokenizer(base_model)
|
199 |
model_neu, device_neu = load_model(model_name, False)
|
200 |
antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,
|
|
|
194 |
#Testen des fine-tuned Modells
|
195 |
print("Predict")
|
196 |
login(token=os.environ["HF_ACCESS_TOKEN"]) #read access token!!!!!!!!!!
|
197 |
+
model_name = "alexkueck/test-tis-1"
|
198 |
tokenizer_neu = load_tokenizer(base_model)
|
199 |
model_neu, device_neu = load_model(model_name, False)
|
200 |
antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,
|